Hugging Face's logo

Shankhdhar
/
classifier_woog

Text Classification
setfit
Safetensors
sentence-transformers
mpnet
generated_from_setfit_trainer
Eval Results (legacy)
Model card Files
xet
Community
Shankhdhar commited on
Commit
ec71f47
·
verified ·
1 Parent(s): 4b10c16

Add SetFit model

Browse files
Files changed (3) hide show
  1. README.md +46 -16
  2. model.safetensors +1 -1
  3. model_head.pkl +1 -1
README.md CHANGED
@@ -32,7 +32,7 @@ model-index:
32
  split: test
33
  metrics:
34
  - type: accuracy
35
- value: 0.8762886597938144
36
  name: Accuracy
37
  ---
38
 
@@ -78,7 +78,7 @@ The model has been trained using an efficient few-shot learning technique that i
78
  ### Metrics
79
  | Label | Accuracy |
80
  |:--------|:---------|
81
- | **all** | 0.8763 |
82
 
83
  ## Uses
84
 
@@ -143,7 +143,7 @@ preds = model("what are the colors available in air jordan 4")
143
 
144
  ### Training Hyperparameters
145
  - batch_size: (16, 16)
146
- - num_epochs: (1, 1)
147
  - max_steps: -1
148
  - sampling_strategy: oversampling
149
  - body_learning_rate: (2e-05, 1e-05)
@@ -162,36 +162,66 @@ preds = model("what are the colors available in air jordan 4")
162
  | Epoch | Step | Training Loss | Validation Loss |
163
  |:------:|:----:|:-------------:|:---------------:|
164
  | 0.0007 | 1 | 0.1501 | - |
165
- | 0.0333 | 50 | 0.1076 | - |
166
- | 0.0667 | 100 | 0.01 | - |
167
- | 0.1 | 150 | 0.0023 | - |
168
- | 0.1333 | 200 | 0.0008 | - |
169
- | 0.1667 | 250 | 0.0007 | - |
170
- | 0.2 | 300 | 0.0005 | - |
171
- | 0.2333 | 350 | 0.0005 | - |
172
- | 0.2667 | 400 | 0.0003 | - |
173
- | 0.3 | 450 | 0.0005 | - |
174
  | 0.3333 | 500 | 0.0003 | - |
175
  | 0.3667 | 550 | 0.0003 | - |
176
  | 0.4 | 600 | 0.0002 | - |
177
  | 0.4333 | 650 | 0.0002 | - |
178
  | 0.4667 | 700 | 0.0003 | - |
179
  | 0.5 | 750 | 0.0002 | - |
180
- | 0.5333 | 800 | 0.0002 | - |
181
  | 0.5667 | 850 | 0.0002 | - |
182
  | 0.6 | 900 | 0.0002 | - |
183
- | 0.6333 | 950 | 0.0002 | - |
184
  | 0.6667 | 1000 | 0.0001 | - |
185
  | 0.7 | 1050 | 0.0001 | - |
186
  | 0.7333 | 1100 | 0.0002 | - |
187
- | 0.7667 | 1150 | 0.0001 | - |
188
  | 0.8 | 1200 | 0.0001 | - |
189
  | 0.8333 | 1250 | 0.0001 | - |
190
- | 0.8667 | 1300 | 0.0002 | - |
191
  | 0.9 | 1350 | 0.0001 | - |
192
  | 0.9333 | 1400 | 0.0002 | - |
193
  | 0.9667 | 1450 | 0.0001 | - |
194
  | 1.0 | 1500 | 0.0002 | - |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
195
 
196
  ### Framework Versions
197
  - Python: 3.9.16
 
32
  split: test
33
  metrics:
34
  - type: accuracy
35
+ value: 0.872852233676976
36
  name: Accuracy
37
  ---
38
 
 
78
  ### Metrics
79
  | Label | Accuracy |
80
  |:--------|:---------|
81
+ | **all** | 0.8729 |
82
 
83
  ## Uses
84
 
 
143
 
144
  ### Training Hyperparameters
145
  - batch_size: (16, 16)
146
+ - num_epochs: (2, 2)
147
  - max_steps: -1
148
  - sampling_strategy: oversampling
149
  - body_learning_rate: (2e-05, 1e-05)
 
162
  | Epoch | Step | Training Loss | Validation Loss |
163
  |:------:|:----:|:-------------:|:---------------:|
164
  | 0.0007 | 1 | 0.1501 | - |
165
+ | 0.0333 | 50 | 0.1476 | - |
166
+ | 0.0667 | 100 | 0.0248 | - |
167
+ | 0.1 | 150 | 0.009 | - |
168
+ | 0.1333 | 200 | 0.0022 | - |
169
+ | 0.1667 | 250 | 0.001 | - |
170
+ | 0.2 | 300 | 0.0009 | - |
171
+ | 0.2333 | 350 | 0.0007 | - |
172
+ | 0.2667 | 400 | 0.0004 | - |
173
+ | 0.3 | 450 | 0.0006 | - |
174
  | 0.3333 | 500 | 0.0003 | - |
175
  | 0.3667 | 550 | 0.0003 | - |
176
  | 0.4 | 600 | 0.0002 | - |
177
  | 0.4333 | 650 | 0.0002 | - |
178
  | 0.4667 | 700 | 0.0003 | - |
179
  | 0.5 | 750 | 0.0002 | - |
180
+ | 0.5333 | 800 | 0.0003 | - |
181
  | 0.5667 | 850 | 0.0002 | - |
182
  | 0.6 | 900 | 0.0002 | - |
183
+ | 0.6333 | 950 | 0.0001 | - |
184
  | 0.6667 | 1000 | 0.0001 | - |
185
  | 0.7 | 1050 | 0.0001 | - |
186
  | 0.7333 | 1100 | 0.0002 | - |
187
+ | 0.7667 | 1150 | 0.0002 | - |
188
  | 0.8 | 1200 | 0.0001 | - |
189
  | 0.8333 | 1250 | 0.0001 | - |
190
+ | 0.8667 | 1300 | 0.0001 | - |
191
  | 0.9 | 1350 | 0.0001 | - |
192
  | 0.9333 | 1400 | 0.0002 | - |
193
  | 0.9667 | 1450 | 0.0001 | - |
194
  | 1.0 | 1500 | 0.0002 | - |
195
+ | 1.0333 | 1550 | 0.0001 | - |
196
+ | 1.0667 | 1600 | 0.0001 | - |
197
+ | 1.1 | 1650 | 0.0001 | - |
198
+ | 1.1333 | 1700 | 0.0001 | - |
199
+ | 1.1667 | 1750 | 0.0001 | - |
200
+ | 1.2 | 1800 | 0.0001 | - |
201
+ | 1.2333 | 1850 | 0.0001 | - |
202
+ | 1.2667 | 1900 | 0.0001 | - |
203
+ | 1.3 | 1950 | 0.0001 | - |
204
+ | 1.3333 | 2000 | 0.0001 | - |
205
+ | 1.3667 | 2050 | 0.0001 | - |
206
+ | 1.4 | 2100 | 0.0001 | - |
207
+ | 1.4333 | 2150 | 0.0001 | - |
208
+ | 1.4667 | 2200 | 0.0001 | - |
209
+ | 1.5 | 2250 | 0.0001 | - |
210
+ | 1.5333 | 2300 | 0.0001 | - |
211
+ | 1.5667 | 2350 | 0.0001 | - |
212
+ | 1.6 | 2400 | 0.0001 | - |
213
+ | 1.6333 | 2450 | 0.0001 | - |
214
+ | 1.6667 | 2500 | 0.0001 | - |
215
+ | 1.7 | 2550 | 0.0001 | - |
216
+ | 1.7333 | 2600 | 0.0001 | - |
217
+ | 1.7667 | 2650 | 0.0001 | - |
218
+ | 1.8 | 2700 | 0.0001 | - |
219
+ | 1.8333 | 2750 | 0.0001 | - |
220
+ | 1.8667 | 2800 | 0.0001 | - |
221
+ | 1.9 | 2850 | 0.0001 | - |
222
+ | 1.9333 | 2900 | 0.0001 | - |
223
+ | 1.9667 | 2950 | 0.0001 | - |
224
+ | 2.0 | 3000 | 0.0001 | - |
225
 
226
  ### Framework Versions
227
  - Python: 3.9.16
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6422d3ea83098d1f7e94cc8aed3489a2e6a3990ee34cdd519a308c98e63356ee
3
  size 437967672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:233fa17e41eedd3308c4817e2a79905d71f3a8039592ff44b6b078f597f18d09
3
  size 437967672
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3731e8ff606284061a188661326d908304d2514b60b7a453926fbb6f55e0f840
3
  size 38311
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f19361422bef33cfeb630852d67b9f643c1ce0ef4b71982d521e05502e7fcba
3
  size 38311