Add SetFit model

1_Pooling/config.json

@@ -5,5 +5,6 @@
   "pooling_mode_max_tokens": false,
   "pooling_mode_mean_sqrt_len_tokens": false,
   "pooling_mode_weightedmean_tokens": false,
-  "pooling_mode_lasttoken": false
+  "pooling_mode_lasttoken": false,
+  "include_prompt": true
 }

README.md

@@ -35,7 +35,7 @@ model-index:
       split: test
     metrics:
     - type: accuracy
-      value: 0.8247011952191236
+      value: 0.8313413014608234
       name: Accuracy
 ---
 
@@ -77,7 +77,7 @@ The model has been trained using an efficient few-shot learning technique that i
 ### Metrics
 | Label   | Accuracy |
 |:--------|:---------|
-| **all** | 0.8247   |
+| **all** | 0.8313   |
 
 ## Uses
 
@@ -95,7 +95,7 @@ Then you can load this model and run inference.
 from setfit import SetFitModel
 
 # Download from the 🤗 Hub
-model = SetFitModel.from_pretrained("setfit_model_id")
+model = SetFitModel.from_pretrained("sultanaw/fine_tuned_setfit_pydata_demo")
 # Run inference
 preds = model("the speakerphone , the radio , all features work perfectly .")
 ```
@@ -156,15 +156,15 @@ preds = model("the speakerphone , the radio , all features work perfectly .")
 ### Training Results
 | Epoch | Step | Training Loss | Validation Loss |
 |:-----:|:----:|:-------------:|:---------------:|
-| 0.1   | 1    | 0.2251        | -               |
+| 0.1   | 1    | 0.2267        | -               |
 
 ### Framework Versions
-- Python: 3.8.10
+- Python: 3.10.12
 - SetFit: 1.0.3
-- Sentence Transformers: 2.3.1
+- Sentence Transformers: 3.0.1
 - Transformers: 4.40.2
-- PyTorch: 2.2.0
-- Datasets: 2.17.1
+- PyTorch: 2.3.0+cu121
+- Datasets: 2.20.0
 - Tokenizers: 0.19.1
 
 ## Citation

config_sentence_transformers.json

@@ -1,7 +1,10 @@
 {
   "__version__": {
-    "sentence_transformers": "2.0.0",
-    "transformers": "4.7.0",
-    "pytorch": "1.9.0+cu102"
-  }
+    "sentence_transformers": "3.0.1",
+    "transformers": "4.40.2",
+    "pytorch": "2.3.0+cu121"
+  },
+  "prompts": {},
+  "default_prompt_name": null,
+  "similarity_fn_name": null
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b6b6068600d9c8c055052a7801d8772885d36d0084f196e5bf090f5f48fb3cca
+oid sha256:9b3973f0a70794e20d19c73b18a6574625bf3787c699c254435d962007d3c089
 size 437967672

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:54a66605c3a689531d714029c59d12b3c76767cd5f5969adaab2f96b36c5f9d1
-size 6991
+oid sha256:32b6cc3c1572cb97d128dc0a026ad2e532d27125c50acd489c3941c41a05b814
+size 7007