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                                                                                                                                                                                   |
-|:------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| 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> |
+| Label | Examples                                                                                                                                                                                                                       |
+|:------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| 0     | <ul><li>'I need to speak to a real person, not a dumb machine.'</li><li>'Stop with the automated nonsense and connect me to a human!'</li><li>'Your automated system is beyond frustrating, let me talk to someone!'</li></ul> |
+| 1     | <ul><li>'I love your new product!'</li><li>'The delivery was very quick!'</li><li>'I would recommend this company to a friend'</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.875  | 10  |
+| Word count   | 4   | 10.0   | 14  |
 
 | Label | Training Sample Count |
 |:------|:----------------------|
-| 0     | 11                    |
-| 1     | 5                     |
+| 0     | 46                    |
+| 1     | 6                     |
 
 ### Training Hyperparameters
 - batch_size: (16, 16)
@@ -150,9 +150,11 @@ preds = model("The product is out of stock")
 - load_best_model_at_end: False
 
 ### Training Results
-| Epoch | Step | Training Loss | Validation Loss |
-|:-----:|:----:|:-------------:|:---------------:|
-| 0.025 | 1    | 0.1106        | -               |
+| Epoch  | Step | Training Loss | Validation Loss |
+|:------:|:----:|:-------------:|:---------------:|
+| 0.0077 | 1    | 0.1479        | -               |
+| 0.3846 | 50   | 0.0008        | -               |
+| 0.7692 | 100  | 0.0005        | -               |
 
 ### Framework Versions
 - Python: 3.11.8

config_setfit.json

@@ -1,4 +1,4 @@
 {
-  "normalize_embeddings": false,
-  "labels": null
+  "labels": null,
+  "normalize_embeddings": false
 }

model.safetensors

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

model_head.pkl

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