Add SetFit model
Browse files- README.md +23 -33
- model.safetensors +1 -1
- model_head.pkl +2 -2
README.md
CHANGED
@@ -9,26 +9,16 @@ base_model: Omar-Nasr/setfitmodel
|
|
9 |
metrics:
|
10 |
- accuracy
|
11 |
widget:
|
12 |
-
- text: ' I
|
13 |
-
|
14 |
-
a
|
15 |
-
- text: '
|
16 |
-
|
17 |
-
- text: '
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
always feel like i walk weird, Have a very asymetrical face (ugly), underdeveloped
|
23 |
-
jaw, i run weird my whole body just doesnt look right'
|
24 |
-
- text: ' In my own experience, I remember I would feel like I was riding high on
|
25 |
-
the crest of a wave of confidence often for days, weeks or even months at a time
|
26 |
-
But it wasn’t until I started to take other positive actions (as outlined in
|
27 |
-
my previous post), that I was able to extend the lengths of my waves of confidence
|
28 |
-
and then break free of the cycle of “highs and lowsâ€_x009d_ altogether But
|
29 |
-
it wasn’t until I started to take other positive actions (as outlined in my
|
30 |
-
previous post), that I was able to extend the lengths of my waves of confidence
|
31 |
-
and then break free of the cycle of “highs and lowsâ€_x009d_ altogether'
|
32 |
pipeline_tag: text-classification
|
33 |
inference: true
|
34 |
model-index:
|
@@ -75,12 +65,12 @@ The model has been trained using an efficient few-shot learning technique that i
|
|
75 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
76 |
|
77 |
### Model Labels
|
78 |
-
| Label | Examples
|
79 |
-
|
80 |
-
| 1
|
81 |
-
| 3
|
82 |
-
| 2
|
83 |
-
| 0
|
84 |
|
85 |
## Evaluation
|
86 |
|
@@ -107,7 +97,7 @@ from setfit import SetFitModel
|
|
107 |
# Download from the 🤗 Hub
|
108 |
model = SetFitModel.from_pretrained("Omar-Nasr/setfitmodel")
|
109 |
# Run inference
|
110 |
-
preds = model("
|
111 |
```
|
112 |
|
113 |
<!--
|
@@ -139,14 +129,14 @@ preds = model(" 18m i messed up at basketball practice today and some teammates
|
|
139 |
### Training Set Metrics
|
140 |
| Training set | Min | Median | Max |
|
141 |
|:-------------|:----|:--------|:----|
|
142 |
-
| Word count |
|
143 |
|
144 |
| Label | Training Sample Count |
|
145 |
|:------|:----------------------|
|
146 |
-
| 0
|
147 |
-
| 1
|
148 |
-
| 2
|
149 |
-
| 3
|
150 |
|
151 |
### Training Hyperparameters
|
152 |
- batch_size: (8, 8)
|
@@ -168,7 +158,7 @@ preds = model(" 18m i messed up at basketball practice today and some teammates
|
|
168 |
### Training Results
|
169 |
| Epoch | Step | Training Loss | Validation Loss |
|
170 |
|:------:|:----:|:-------------:|:---------------:|
|
171 |
-
| 0.0417 | 1 | 0.
|
172 |
|
173 |
### Framework Versions
|
174 |
- Python: 3.10.13
|
|
|
9 |
metrics:
|
10 |
- accuracy
|
11 |
widget:
|
12 |
+
- text: ' I didn''t go outside too much'
|
13 |
+
- text: ' I like art and nature but you can’t really talk about those for more than
|
14 |
+
a few seconds'
|
15 |
+
- text: ' The only times I call in is when I am so sick that I can''t physically walk
|
16 |
+
On Monday, I was very sick and running a 101'
|
17 |
+
- text: ' But even then, I didn''t have any other problems outside school I still
|
18 |
+
had no friends at European school, I haven''t had any walks which I had constantly
|
19 |
+
with my friends back in Ukraine'
|
20 |
+
- text: ' Your body is trying to reduce weight by throwing up and having to go to
|
21 |
+
the bathroon, in case you need to run from the "enemy", so you''ll be lighter'
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
22 |
pipeline_tag: text-classification
|
23 |
inference: true
|
24 |
model-index:
|
|
|
65 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
66 |
|
67 |
### Model Labels
|
68 |
+
| Label | Examples |
|
69 |
+
|:------|:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
70 |
+
| 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> |
|
71 |
+
| 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> |
|
72 |
+
| 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> |
|
73 |
+
| 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> |
|
74 |
|
75 |
## Evaluation
|
76 |
|
|
|
97 |
# Download from the 🤗 Hub
|
98 |
model = SetFitModel.from_pretrained("Omar-Nasr/setfitmodel")
|
99 |
# Run inference
|
100 |
+
preds = model(" I didn't go outside too much")
|
101 |
```
|
102 |
|
103 |
<!--
|
|
|
129 |
### Training Set Metrics
|
130 |
| Training set | Min | Median | Max |
|
131 |
|:-------------|:----|:--------|:----|
|
132 |
+
| Word count | 7 | 33.0625 | 87 |
|
133 |
|
134 |
| Label | Training Sample Count |
|
135 |
|:------|:----------------------|
|
136 |
+
| 0.0 | 4 |
|
137 |
+
| 1.0 | 4 |
|
138 |
+
| 2.0 | 4 |
|
139 |
+
| 3.0 | 4 |
|
140 |
|
141 |
### Training Hyperparameters
|
142 |
- batch_size: (8, 8)
|
|
|
158 |
### Training Results
|
159 |
| Epoch | Step | Training Loss | Validation Loss |
|
160 |
|:------:|:----:|:-------------:|:---------------:|
|
161 |
+
| 0.0417 | 1 | 0.0033 | - |
|
162 |
|
163 |
### Framework Versions
|
164 |
- Python: 3.10.13
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 1421483904
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a30000c458cfbf1099d2d1d02cff10cf7ead86d58820d61423625ae0fb8380f5
|
3 |
size 1421483904
|
model_head.pkl
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:beba8d1ca041631a7df9a2906e61fff1ee970713a6c2c6faecd7161645a8334c
|
3 |
+
size 33639
|