Add SetFit model

README.md

@@ -62,10 +62,10 @@ The model has been trained using an efficient few-shot learning technique that i
 - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
 
 ### Model Labels
-| Label | Examples                                                                                                                                                  |
-|:------|:----------------------------------------------------------------------------------------------------------------------------------------------------------|
-| 0     | <ul><li>'I received the wrong item'</li><li>'The product arrived damaged'</li><li>"I placed an order but haven't received a confirmation email"</li></ul> |
-| 1     | <ul><li>'I got a discount code!'</li><li>'I would recommend this company to a friend'</li><li>"I'm very satisfied with my purchase"</li></ul>             |
+| Label | Examples                                                                                                                                                                                   |
+|:------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| 1     | <ul><li>'I got a discount code!'</li><li>'I would recommend this company to a friend'</li><li>'The return process was easy'</li></ul>                                                      |
+| 0     | <ul><li>'You are not understanding what i am saying'</li><li>"I'm having trouble logging in to my account"</li><li>"I placed an order but haven't received a confirmation email"</li></ul> |
 
 ## Evaluation
 
@@ -124,12 +124,12 @@ preds = model("The product is out of stock")
 ### Training Set Metrics
 | Training set | Min | Median | Max |
 |:-------------|:----|:-------|:----|
-| Word count   | 4   | 6.9375 | 10  |
+| Word count   | 4   | 6.875  | 10  |
 
 | Label | Training Sample Count |
 |:------|:----------------------|
-| 0     | 10                    |
-| 1     | 6                     |
+| 0     | 11                    |
+| 1     | 5                     |
 
 ### Training Hyperparameters
 - batch_size: (16, 16)
@@ -152,7 +152,7 @@ preds = model("The product is out of stock")
 ### Training Results
 | Epoch | Step | Training Loss | Validation Loss |
 |:-----:|:----:|:-------------:|:---------------:|
-| 0.025 | 1    | 0.0549        | -               |
+| 0.025 | 1    | 0.1106        | -               |
 
 ### Framework Versions
 - Python: 3.11.8

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f371e87d5d8ba17bd19ce9e245b362cf1074fe0b544a79ad164520a0d27fea7f
+oid sha256:11f5f369329579350ce85bc4ad09977f481c979db950823b9dd40a4eaa3a7358
 size 265462608

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b0d861bdc121e04180575e968b947df9a4244d3e69030d5a39e058505013e1d5
+oid sha256:960f91ea38d617fea8e55beccde57301118285a9427d4c7deb445ac0263ccdbc
 size 7013