Hugging Face's logo

mann2107
/
BCMPIIRAB_MiniLM_ALLNew

Text Classification
setfit
Safetensors
sentence-transformers
bert
generated_from_setfit_trainer
Model card Files Community
mann2107 commited on
Commit
b387a95
•
1 Parent(s): 103aa03

Push model using huggingface_hub.

Browse files
Files changed (5) hide show
  1. README.md +91 -8
  2. config.json +1 -1
  3. config_setfit.json +2 -2
  4. model.safetensors +1 -1
  5. model_head.pkl +1 -1
README.md CHANGED
@@ -140,13 +140,13 @@ preds = model("Thank you for your email. Please go ahead and issue. Please invoi
140
  | 13 | 24 |
141
 
142
  ### Training Hyperparameters
143
- - batch_size: (64, 64)
144
  - num_epochs: (1, 1)
145
  - max_steps: -1
146
  - sampling_strategy: oversampling
147
- - num_iterations: 2
148
- - body_learning_rate: (0.00012410271291092199, 0.00012410271291092199)
149
- - head_learning_rate: 0.0001551076479757721
150
  - loss: CosineSimilarityLoss
151
  - distance_metric: cosine_distance
152
  - margin: 0.25
@@ -159,10 +159,93 @@ preds = model("Thank you for your email. Please go ahead and issue. Please invoi
159
  - load_best_model_at_end: True
160
 
161
  ### Training Results
162
- | Epoch | Step | Training Loss | Validation Loss |
163
- |:-------:|:------:|:-------------:|:---------------:|
164
- | 0.0476 | 1 | 0.2484 | - |
165
- | **1.0** | **21** | **-** | **0.1014** |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
166
 
167
  * The bold row denotes the saved checkpoint.
168
  ### Framework Versions
 
140
  | 13 | 24 |
141
 
142
  ### Training Hyperparameters
143
+ - batch_size: (16, 16)
144
  - num_epochs: (1, 1)
145
  - max_steps: -1
146
  - sampling_strategy: oversampling
147
+ - num_iterations: 99
148
+ - body_learning_rate: (0.0002733656643765287, 0.0002733656643765287)
149
+ - head_learning_rate: 2.7029049129688732e-05
150
  - loss: CosineSimilarityLoss
151
  - distance_metric: cosine_distance
152
  - margin: 0.25
 
159
  - load_best_model_at_end: True
160
 
161
  ### Training Results
162
+ | Epoch | Step | Training Loss | Validation Loss |
163
+ |:-------:|:--------:|:-------------:|:---------------:|
164
+ | 0.0002 | 1 | 0.2546 | - |
165
+ | 0.0120 | 50 | 0.1667 | - |
166
+ | 0.0241 | 100 | 0.1165 | - |
167
+ | 0.0361 | 150 | 0.0799 | - |
168
+ | 0.0481 | 200 | 0.0212 | - |
169
+ | 0.0601 | 250 | 0.0188 | - |
170
+ | 0.0722 | 300 | 0.0531 | - |
171
+ | 0.0842 | 350 | 0.0273 | - |
172
+ | 0.0962 | 400 | 0.0111 | - |
173
+ | 0.1082 | 450 | 0.0203 | - |
174
+ | 0.1203 | 500 | 0.0397 | - |
175
+ | 0.1323 | 550 | 0.0164 | - |
176
+ | 0.1443 | 600 | 0.0045 | - |
177
+ | 0.1563 | 650 | 0.0032 | - |
178
+ | 0.1684 | 700 | 0.001 | - |
179
+ | 0.1804 | 750 | 0.0011 | - |
180
+ | 0.1924 | 800 | 0.0004 | - |
181
+ | 0.2044 | 850 | 0.0009 | - |
182
+ | 0.2165 | 900 | 0.0006 | - |
183
+ | 0.2285 | 950 | 0.0008 | - |
184
+ | 0.2405 | 1000 | 0.0004 | - |
185
+ | 0.2525 | 1050 | 0.0008 | - |
186
+ | 0.2646 | 1100 | 0.0005 | - |
187
+ | 0.2766 | 1150 | 0.0006 | - |
188
+ | 0.2886 | 1200 | 0.0007 | - |
189
+ | 0.3006 | 1250 | 0.0043 | - |
190
+ | 0.3127 | 1300 | 0.0004 | - |
191
+ | 0.3247 | 1350 | 0.0005 | - |
192
+ | 0.3367 | 1400 | 0.0005 | - |
193
+ | 0.3487 | 1450 | 0.0004 | - |
194
+ | 0.3608 | 1500 | 0.0004 | - |
195
+ | 0.3728 | 1550 | 0.0005 | - |
196
+ | 0.3848 | 1600 | 0.0007 | - |
197
+ | 0.3968 | 1650 | 0.0006 | - |
198
+ | 0.4089 | 1700 | 0.0002 | - |
199
+ | 0.4209 | 1750 | 0.0006 | - |
200
+ | 0.4329 | 1800 | 0.0008 | - |
201
+ | 0.4449 | 1850 | 0.0003 | - |
202
+ | 0.4570 | 1900 | 0.0005 | - |
203
+ | 0.4690 | 1950 | 0.0003 | - |
204
+ | 0.4810 | 2000 | 0.0003 | - |
205
+ | 0.4930 | 2050 | 0.0003 | - |
206
+ | 0.5051 | 2100 | 0.0006 | - |
207
+ | 0.5171 | 2150 | 0.0003 | - |
208
+ | 0.5291 | 2200 | 0.0002 | - |
209
+ | 0.5411 | 2250 | 0.0002 | - |
210
+ | 0.5532 | 2300 | 0.0002 | - |
211
+ | 0.5652 | 2350 | 0.0004 | - |
212
+ | 0.5772 | 2400 | 0.0003 | - |
213
+ | 0.5892 | 2450 | 0.0003 | - |
214
+ | 0.6013 | 2500 | 0.0002 | - |
215
+ | 0.6133 | 2550 | 0.0002 | - |
216
+ | 0.6253 | 2600 | 0.0013 | - |
217
+ | 0.6373 | 2650 | 0.0002 | - |
218
+ | 0.6494 | 2700 | 0.0007 | - |
219
+ | 0.6614 | 2750 | 0.0004 | - |
220
+ | 0.6734 | 2800 | 0.0007 | - |
221
+ | 0.6854 | 2850 | 0.0018 | - |
222
+ | 0.6975 | 2900 | 0.0002 | - |
223
+ | 0.7095 | 2950 | 0.0003 | - |
224
+ | 0.7215 | 3000 | 0.0006 | - |
225
+ | 0.7335 | 3050 | 0.0003 | - |
226
+ | 0.7456 | 3100 | 0.0002 | - |
227
+ | 0.7576 | 3150 | 0.0002 | - |
228
+ | 0.7696 | 3200 | 0.0002 | - |
229
+ | 0.7816 | 3250 | 0.0002 | - |
230
+ | 0.7937 | 3300 | 0.0002 | - |
231
+ | 0.8057 | 3350 | 0.0001 | - |
232
+ | 0.8177 | 3400 | 0.0003 | - |
233
+ | 0.8297 | 3450 | 0.0002 | - |
234
+ | 0.8418 | 3500 | 0.0002 | - |
235
+ | 0.8538 | 3550 | 0.0002 | - |
236
+ | 0.8658 | 3600 | 0.0002 | - |
237
+ | 0.8778 | 3650 | 0.0002 | - |
238
+ | 0.8899 | 3700 | 0.0002 | - |
239
+ | 0.9019 | 3750 | 0.0005 | - |
240
+ | 0.9139 | 3800 | 0.0002 | - |
241
+ | 0.9259 | 3850 | 0.0001 | - |
242
+ | 0.9380 | 3900 | 0.0004 | - |
243
+ | 0.9500 | 3950 | 0.0001 | - |
244
+ | 0.9620 | 4000 | 0.0005 | - |
245
+ | 0.9740 | 4050 | 0.0002 | - |
246
+ | 0.9861 | 4100 | 0.0002 | - |
247
+ | 0.9981 | 4150 | 0.0001 | - |
248
+ | **1.0** | **4158** | **-** | **0.0302** |
249
 
250
  * The bold row denotes the saved checkpoint.
251
  ### Framework Versions
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "checkpoints/step_21",
3
  "architectures": [
4
  "BertModel"
5
  ],
 
1
  {
2
+ "_name_or_path": "checkpoints/step_4158",
3
  "architectures": [
4
  "BertModel"
5
  ],
config_setfit.json CHANGED
@@ -1,4 +1,4 @@
1
  {
2
- "labels": null,
3
- "normalize_embeddings": false
4
  }
 
1
  {
2
+ "normalize_embeddings": false,
3
+ "labels": null
4
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5bbc2bd8d39472e6d55398e4a8426eea02aac2568fa819bf58301211bd62305e
3
  size 90864192
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e337745bca23266645ad46209eaa4050887dc0691ce3e52fe8e3bf85b0e5272
3
  size 90864192
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c0b48c9fb9b7381fc9b0fcf3195187185f6e455b747f94651270e9e6d587ebd3
3
  size 44071
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d6be24185ee00c4af91881cdf3cd116145a8eb7254e4e39b7460fdd48e12ca7
3
  size 44071