Hugging Face's logo

research-dump
/
bge-small-en-v1.5_wikinews_outcome_prediction_v1

Text Classification
setfit
Safetensors
sentence-transformers
bert
generated_from_setfit_trainer
Model card Files Community
hsuvaskakoty commited on
Commit
5a8e2bd
•
1 Parent(s): aa7d3a3

Push model using huggingface_hub.

Browse files
Files changed (4) hide show
  1. README.md +17 -17
  2. config_setfit.json +2 -2
  3. model.safetensors +1 -1
  4. model_head.pkl +1 -1
README.md CHANGED
@@ -203,8 +203,8 @@ preds = model(" Interviews: Interviews WN:Story preparation/Office hours:Frank
203
  - max_steps: -1
204
  - sampling_strategy: oversampling
205
  - num_iterations: 100
206
- - body_learning_rate: (0.001, 0.001)
207
- - head_learning_rate: 1e-05
208
  - loss: CosineSimilarityLoss
209
  - distance_metric: cosine_distance
210
  - margin: 0.25
@@ -220,21 +220,21 @@ preds = model(" Interviews: Interviews WN:Story preparation/Office hours:Frank
220
  | Epoch | Step | Training Loss | Validation Loss |
221
  |:------:|:----:|:-------------:|:---------------:|
222
  | 0.0006 | 1 | 0.2247 | - |
223
- | 0.0635 | 100 | 0.14 | 0.3160 |
224
- | 0.1270 | 200 | 0.3128 | 0.5 |
225
- | 0.1905 | 300 | 0.2701 | 0.5 |
226
- | 0.2540 | 400 | 0.2566 | 0.5 |
227
- | 0.3175 | 500 | 0.2601 | 0.5 |
228
- | 0.3810 | 600 | 0.2557 | 0.5 |
229
- | 0.4444 | 700 | 0.2556 | 0.5 |
230
- | 0.5079 | 800 | 0.254 | 0.5000 |
231
- | 0.5714 | 900 | 0.2538 | 0.5 |
232
- | 0.6349 | 1000 | 0.2546 | 0.5 |
233
- | 0.6984 | 1100 | 0.2517 | 0.5 |
234
- | 0.7619 | 1200 | 0.2554 | 0.5 |
235
- | 0.8254 | 1300 | 0.2542 | 0.5 |
236
- | 0.8889 | 1400 | 0.2491 | 0.5 |
237
- | 0.9524 | 1500 | 0.2546 | 0.5 |
238
 
239
  ### Framework Versions
240
  - Python: 3.10.12
 
203
  - max_steps: -1
204
  - sampling_strategy: oversampling
205
  - num_iterations: 100
206
+ - body_learning_rate: (0.0001, 0.0001)
207
+ - head_learning_rate: 0.0004
208
  - loss: CosineSimilarityLoss
209
  - distance_metric: cosine_distance
210
  - margin: 0.25
 
220
  | Epoch | Step | Training Loss | Validation Loss |
221
  |:------:|:----:|:-------------:|:---------------:|
222
  | 0.0006 | 1 | 0.2247 | - |
223
+ | 0.0635 | 100 | 0.1907 | 0.2608 |
224
+ | 0.1270 | 200 | 0.0381 | 0.3005 |
225
+ | 0.1905 | 300 | 0.0118 | 0.3670 |
226
+ | 0.2540 | 400 | 0.0022 | 0.3018 |
227
+ | 0.3175 | 500 | 0.0009 | 0.3258 |
228
+ | 0.3810 | 600 | 0.0007 | 0.3361 |
229
+ | 0.4444 | 700 | 0.0006 | 0.3460 |
230
+ | 0.5079 | 800 | 0.0006 | 0.3637 |
231
+ | 0.5714 | 900 | 0.0005 | 0.3850 |
232
+ | 0.6349 | 1000 | 0.0004 | 0.3820 |
233
+ | 0.6984 | 1100 | 0.0004 | 0.3902 |
234
+ | 0.7619 | 1200 | 0.0004 | 0.4078 |
235
+ | 0.8254 | 1300 | 0.0004 | 0.4081 |
236
+ | 0.8889 | 1400 | 0.0004 | 0.4124 |
237
+ | 0.9524 | 1500 | 0.0004 | 0.4120 |
238
 
239
  ### Framework Versions
240
  - Python: 3.10.12
config_setfit.json CHANGED
@@ -1,4 +1,4 @@
1
  {
2
- "normalize_embeddings": false,
3
- "labels": null
4
  }
 
1
  {
2
+ "labels": null,
3
+ "normalize_embeddings": false
4
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:570447cbc0761018c7edafe6d9e8eb1cb2f6e8ce4d14c2e526c6480da2e6504e
3
  size 133462128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:842fb152697da33016c46a5f86e816a1caa08e9e4beb7bf16fd0bb9c9ba966ee
3
  size 133462128
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:270a34760870d9489606312675e973e70b75caf0352b1c609751b7ef3fb17e34
3
  size 13175
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:836cca13d9f8a4837a107eab4fc7e55430752c8be539c7760b1ed0761b1c4e74
3
  size 13175