Add SetFit model

README.md

@@ -9,16 +9,17 @@ base_model: Omar-Nasr/setfitmodel
 metrics:
 - accuracy
 widget:
-- text: ' I didn''t go outside too much'
 - text: ' I like art and nature but you can’t really talk about those for more than
     a few seconds'
-- text: ' The only times I call in is when I am so sick that I can''t physically walk
-    On Monday, I was very sick and running a 101'
-- text: ' But even then, I didn''t have any other problems outside school I still
-    had no friends at European school, I haven''t had any walks which I had constantly
-    with my friends back in Ukraine'
-- text: ' Your body is trying to reduce weight by throwing up and having to go to
-    the bathroon, in case you need to run from the "enemy", so you''ll be lighter'
+- text: ' That''s kind of the nature of my volunteer work, but you could volunteer
+    with a food bank or boys  and girls club, which would involve more social interaction
+    Just breaking that cycle by going for a short walk around the neighbourhood is
+    a good idea'
+- text: ' That being heat, sweat, more people outside and you wear less clothes on
+    you (not so comfortable being a fat guy and sweat can be seen on your shirt)'
+- text: ' Want to join soccer club but so scared'
+- text: ' I literally do not leave the house, I will sometimes go in the garden but
+    no further So I just ran outside and back to the car'
 pipeline_tag: text-classification
 inference: true
 model-index:
@@ -33,7 +34,7 @@ model-index:
       split: test
     metrics:
     - type: accuracy
-      value: 0.49166666666666664
+      value: 0.5866666666666667
       name: Accuracy
 ---
 
@@ -65,19 +66,19 @@ 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.0   | <ul><li>' Try challenging yourself more, take a walk in the park, small things like that make you better  Try challenging yourself more, take a walk in the park, small things like that make you better '</li><li>' Last year most of my college friends took advantage of Labor Day weekend to go on a trip to a cabin and have fun together Last year most of my college friends took advantage of Labor Day weekend to go on a trip to a cabin and have fun together I was also thinking about going on a backpacking trip alone so I can get some time to myself, because I don’t like how my relationships with other people are affecting my sense of self worth'</li><li>' and i do biking, i bike for atleast 20mins and towards the end i feel anxious free, less socially anxious too and i do biking, i bike for atleast 20mins and towards the end i feel anxious free, less socially anxious too'</li></ul> |
-| 3.0   | <ul><li>"It's whenever I look someone in the eye, when I'm on my bike or when I'm walking somewhere, I feel like they look at me and ridicule me in their headsIt's whenever I look someone in the eye, when I'm on my bike or when I'm walking somewhere, I feel like they look at me and ridicule me in their heads"</li><li>" But even then, I didn't have any other problems outside school I still had no friends at European school, I haven't had any walks which I had constantly with my friends back in Ukraine"</li><li>' However, one day after lunch I was walking to his classroom to get to his class Every time I go outside I instantly get nervous'</li></ul>                                                                                                                                                                                                                                           |
-| 2.0   | <ul><li>' I literally do not leave the house, I will sometimes go in the garden but no further So I just ran outside and back to the car'</li><li>" But we can't sleep in a hotel, we have go camping"</li><li>' I like art and nature but you can’t really talk about those for more than a few seconds'</li></ul>                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                       |
-| 0.0   | <ul><li>" I'm not a fat, ugly, 400 lb guy who doesn't shower, I'm average looking, maybe a little overweight for my height, kind of short cause it runs in my family"</li><li>' I would do this too but then I run the risk of meeting someone I know lol'</li><li>" The only times I call in is when I am so sick that I can't physically walk On Monday, I was very sick and running a 101"</li></ul>                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                   |
+| Label | Examples                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                               |
+|:------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| 1.0   | <ul><li>' Go out for a walk once a day additionally and slowly start increasing the time you spend outside Go out for a walk once a day additionally and slowly start increasing the time you spend outside Start doing sport, either outdoors or at a gym If you can, try to take your dog to a dog park or something like that'</li><li>' Try challenging yourself more, take a walk in the park, small things like that make you better  Try challenging yourself more, take a walk in the park, small things like that make you better '</li><li>" Now I'm not saying to go to a party on the spot, just go out, shop, take a walk in the park, that kind of thing Now I'm not saying to go to a party on the spot, just go out, shop, take a walk in the park, that kind of thing"</li></ul>                                                      |
+| 2.0   | <ul><li>' I’m an equestrian, so I ride horses and manage for a pretty famous trainer I can hold a non work related conversation with a stranger while I’m working but if I met that same person outside of the work day I’d have a panic attack and not be able to say a word'</li><li>' On long walks to errands, and whilst power walking for exercise'</li><li>' She said no, but we have a tasty forest fruit mix cake I felt high as a kite walking home'</li></ul>                                                                                                                                                                                                                                                                                                                                                                               |
+| 0.0   | <ul><li>' Good to know that some people are in the same camp'</li><li>" I'm sure if the worlds ever did clash that your friends would understand (few people actually enjoy being at work) and, worst case scenario, your coworkers would be surprised at your outgoing nature while around friends"</li><li>' If anything you should be thinking about wearing sun screen so you retain your good skin as it becomes your ally as you age outside'</li></ul>                                                                                                                                                                                                                                                                                                                                                                                          |
+| 3.0   | <ul><li>" While I ended up making progress, it wasn't as fast as I had hoped and I still had a lot of trouble doing some things (such as jogging in public)"</li><li>' One, frack you guys who say “just get over it”, you’ve probably never dealt with anxiety, it’s like you are carrying the weight of everyone’s judgements and eyes on you with every possibility of any and every event running through your head all the time I am trying, I force myself outside and to interact but it’s terrifying and people just don’t seem to get that'</li><li>" I want to go swimming, anxiety and low self esteem make it really hard I want to go swimming, anxiety and low self esteem make it really hard I'm at least planning to go for a swim at a nearby lake but there is one problem I have: I'm not really confident with my body"</li></ul> |
 
 ## Evaluation
 
 ### Metrics
 | Label   | Accuracy |
 |:--------|:---------|
-| **all** | 0.4917   |
+| **all** | 0.5867   |
 
 ## Uses
 
@@ -97,7 +98,7 @@ from setfit import SetFitModel
 # Download from the 🤗 Hub
 model = SetFitModel.from_pretrained("Omar-Nasr/setfitmodel")
 # Run inference
-preds = model(" I didn't go outside too much")
+preds = model(" Want to join soccer club but so scared")
 ```
 
 <!--
@@ -127,16 +128,16 @@ preds = model(" I didn't go outside too much")
 ## Training Details
 
 ### Training Set Metrics
-| Training set | Min | Median  | Max |
-|:-------------|:----|:--------|:----|
-| Word count   | 7   | 33.0625 | 87  |
+| Training set | Min | Median  | Max  |
+|:-------------|:----|:--------|:-----|
+| Word count   | 4   | 51.2656 | 1083 |
 
 | Label | Training Sample Count |
 |:------|:----------------------|
-| 0.0   | 4                     |
-| 1.0   | 4                     |
-| 2.0   | 4                     |
-| 3.0   | 4                     |
+| 0.0   | 16                    |
+| 1.0   | 16                    |
+| 2.0   | 16                    |
+| 3.0   | 16                    |
 
 ### Training Hyperparameters
 - batch_size: (8, 8)
@@ -158,7 +159,14 @@ preds = model(" I didn't go outside too much")
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.0417 | 1    | 0.0033        | -               |
+| 0.0026 | 1    | 0.0           | -               |
+| 0.1302 | 50   | 0.0001        | -               |
+| 0.2604 | 100  | 0.0           | -               |
+| 0.3906 | 150  | 0.0           | -               |
+| 0.5208 | 200  | 0.0           | -               |
+| 0.6510 | 250  | 0.0           | -               |
+| 0.7812 | 300  | 0.0           | -               |
+| 0.9115 | 350  | 0.0           | -               |
 
 ### Framework Versions
 - Python: 3.10.13

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a30000c458cfbf1099d2d1d02cff10cf7ead86d58820d61423625ae0fb8380f5
+oid sha256:47d5829d51b2b3335f5b31b2902c5654da96b96aeae75a73035e4d3e34eca94b
 size 1421483904

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:beba8d1ca041631a7df9a2906e61fff1ee970713a6c2c6faecd7161645a8334c
+oid sha256:5675756feed913ca0a3ac1669818481972b7f69bcbb0c404374c09b2e4d10c98
 size 33639