Add SetFit model

README.md

@@ -8,18 +8,26 @@ tags:
 metrics:
 - accuracy
 widget:
-- text: She is Female, her heart rate is 89, she walks 3873 steps daily and is Overweight.
-    She slept at 2 hrs. Yesterday, she slept from 4.0 hrs to 6.0 hrs, with a duration
-    of 120.0 minutes and 1 interruptions. The day before yesterday, she slept from
-    4.0 hrs to 9.0 hrs, with a duration of 300.0 minutes and 2 interruptions.
-- text: She is Female, her heart rate is 66, she walks 2413 steps daily and is Underweight.
-    She slept at 23 hrs. Yesterday, she slept from 23.0 hrs to 7.0 hrs, with a duration
-    of 472.0 minutes and 5 interruptions. The day before yesterday, she slept from
-    23.0 hrs to 5.0 hrs, with a duration of 344.0 minutes and 6 interruptions.
-- text: He is Male, his heart rate is 64, he walks 10000 steps daily, and is Normal.
-    He slept at 11 hrs. Yesterday, he slept from 22.0hrs to 11.0 hrs, with a duration
-    of 765.0 minutes and 2 interruptions. The day before yesterday, he slept from
-    23.0 hrs to 8.0 hrs, with a duration of 527.0 minutes and 4 interruptions.
+- text: She is Female, her heart rate is 63, she walks 4000 steps daily and is Underweight.
+    She slept at 2 hrs. Yesterday, she slept from 1 hrs to 7 hrs, with a duration
+    of 360 minutes and 5 interruptions. The day before yesterday, she slept from 23
+    hrs to 7 hrs, with a duration of 420 minutes and 3 interruptions.
+- text: She is Female, her heart rate is 70, she walks 8000 steps daily and is Normal.
+    She slept at 22 hrs. Yesterday, she slept from 23 hrs to 7 hrs, with a duration
+    of 400 minutes and 2 interruptions. The day before yesterday, she slept from 22
+    hrs to 6 hrs, with a duration of 430 minutes and 2 interruptions.
+- text: He is Male, his heart rate is 70, he walks 2400 steps daily, and is Underweight.
+    He slept at 0 hrs. Yesterday, he slept from 2hrs to 7 hrs, with a duration of
+    280 minutes and 4 interruptions. The day before yesterday, he slept from 2 hrs
+    to 8 hrs, with a duration of 340 minutes and 4 interruptions.
+- text: She is Female, her heart rate is 68, she walks 11,000 steps daily and is Normal.
+    She slept at 1 hrs. Yesterday, she slept from 1 hrs to 9 hrs, with a duration
+    of 495 minutes and 0 interruptions. The day before yesterday, she slept from 1
+    hrs to 10 hrs, with a duration of 540 minutes and 1 interruptions.
+- text: He is Male, his heart rate is 67, he walks 12000 steps daily, and is Normal.
+    He slept at 3 hrs. Yesterday, he slept from 4hrs to 11 hrs, with a duration of
+    420 minutes and 3 interruptions. The day before yesterday, he slept from 3 hrs
+    to 5 hrs, with a duration of 150 minutes and 0 interruptions.
 pipeline_tag: text-classification
 inference: false
 base_model: sentence-transformers/paraphrase-mpnet-base-v2
@@ -35,7 +43,7 @@ model-index:
       split: test
     metrics:
     - type: accuracy
-      value: 0.3333333333333333
+      value: 0.0
       name: Accuracy
 ---
 
@@ -71,7 +79,7 @@ The model has been trained using an efficient few-shot learning technique that i
 ### Metrics
 | Label   | Accuracy |
 |:--------|:---------|
-| **all** | 0.3333   |
+| **all** | 0.0      |
 
 ## Uses
 
@@ -91,7 +99,7 @@ from setfit import SetFitModel
 # Download from the 🤗 Hub
 model = SetFitModel.from_pretrained("naushin/setfit-ethos-multilabel-example")
 # Run inference
-preds = model("He is Male, his heart rate is 64, he walks 10000 steps daily, and is Normal. He slept at 11 hrs. Yesterday, he slept from 22.0hrs to 11.0 hrs, with a duration of 765.0 minutes and 2 interruptions. The day before yesterday, he slept from 23.0 hrs to 8.0 hrs, with a duration of 527.0 minutes and 4 interruptions.")
+preds = model("He is Male, his heart rate is 67, he walks 12000 steps daily, and is Normal. He slept at 3 hrs. Yesterday, he slept from 4hrs to 11 hrs, with a duration of 420 minutes and 3 interruptions. The day before yesterday, he slept from 3 hrs to 5 hrs, with a duration of 150 minutes and 0 interruptions.")
 ```
 
 <!--
@@ -144,9 +152,9 @@ preds = model("He is Male, his heart rate is 64, he walks 10000 steps daily, and
 - load_best_model_at_end: False
 
 ### Training Results
-| Epoch | Step | Training Loss | Validation Loss |
-|:-----:|:----:|:-------------:|:---------------:|
-| 0.1   | 1    | 0.4129        | -               |
+| Epoch  | Step | Training Loss | Validation Loss |
+|:------:|:----:|:-------------:|:---------------:|
+| 0.0667 | 1    | 0.421         | -               |
 
 ### Framework Versions
 - Python: 3.10.12

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:72ca11176130d67d0846a7ac7b816e87c8746228f049f51ff23bf89bbe676444
+oid sha256:670797c939640a61f251901631eef747ef0a8846b48e27cd9ee308bcf3209a34
 size 437967672

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:80f9a59e0587f17932f8129ce57738cb2e436b747b44a46b163a8bd2c7f7e3c9
+oid sha256:7e07d4db9bf4602cd900a953631193fad37e32335ff96347071599d9217546e6
 size 20436