Add SetFit model

README.md

@@ -8,6 +8,9 @@ tags:
 base_model: sentence-transformers/all-mpnet-base-v2
 metrics:
 - accuracy
+- precision
+- recall
+- f1
 widget:
 - text: I appreciate the smooth operation of our public transit system every day.
 - text: Precision shines through in your flawless execution of the project.
@@ -31,6 +34,15 @@ model-index:
     - type: accuracy
       value: 0.5
       name: Accuracy
+    - type: precision
+      value: 0.8
+      name: Precision
+    - type: recall
+      value: 0.8888888888888888
+      name: Recall
+    - type: f1
+      value: 0.8421052631578947
+      name: F1
 ---
 
 # SetFit with sentence-transformers/all-mpnet-base-v2
@@ -63,9 +75,9 @@ The model has been trained using an efficient few-shot learning technique that i
 ## Evaluation
 
 ### Metrics
-| Label   | Accuracy |
-|:--------|:---------|
-| **all** | 0.5      |
+| Label   | Accuracy | Precision | Recall | F1     |
+|:--------|:---------|:----------|:-------|:-------|
+| **all** | 0.5      | 0.8       | 0.8889 | 0.8421 |
 
 ## Uses
 
@@ -143,15 +155,15 @@ preds = model("Well done on orchestrating such a seamless event!")
 | 1.0     | 13     | -             | 0.1928          |
 | 2.0     | 26     | -             | 0.1831          |
 | 3.0     | 39     | -             | 0.1724          |
-| 3.8462  | 50     | 0.0796        | -               |
-| **4.0** | **52** | **-**         | **0.1615**      |
-| 5.0     | 65     | -             | 0.1698          |
-| 6.0     | 78     | -             | 0.1842          |
-| 7.0     | 91     | -             | 0.1909          |
+| 3.8462  | 50     | 0.08          | -               |
+| **4.0** | **52** | **-**         | **0.1614**      |
+| 5.0     | 65     | -             | 0.1695          |
+| 6.0     | 78     | -             | 0.1837          |
+| 7.0     | 91     | -             | 0.1904          |
 | 7.6923  | 100    | 0.0364        | -               |
-| 8.0     | 104    | -             | 0.2004          |
-| 9.0     | 117    | -             | 0.1999          |
-| 10.0    | 130    | -             | 0.1973          |
+| 8.0     | 104    | -             | 0.1997          |
+| 9.0     | 117    | -             | 0.1994          |
+| 10.0    | 130    | -             | 0.1967          |
 
 * The bold row denotes the saved checkpoint.
 ### Framework Versions

config_setfit.json

@@ -1,8 +1,8 @@
 {
-  "normalize_embeddings": false,
   "labels": [
     "Admiration",
     "Gratitude",
     "Praise"
-  ]
+  ],
+  "normalize_embeddings": false
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:324e141ba908f82021ce24e1fae494155ad3a9042ae5ecdb17fb03e65a828ab4
+oid sha256:d1a93e67feec2184b1995cf79c3a0b0cfce0d2902533007d9d0ac537391cdf07
 size 437967672

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e723f9bda1bbf8f9d37fa725d8c2b5ba10451cfe3506d15218dffe76842e6e24
+oid sha256:27e22f3d6674011f92cc257315727a60fb0d41a24a338412f4d57bec52cbca85
 size 20321