Hugging Face's logo

mann2107
/
BCMPIIRAB_MiniLM_ALL

Text Classification
setfit
Safetensors
sentence-transformers
bert
generated_from_setfit_trainer
Model card Files Community
mann2107 commited on
Commit
7e705ef
•
1 Parent(s): 8b16318

Push model using huggingface_hub.

Browse files
Files changed (5) hide show
  1. README.md +11 -7
  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
@@ -144,24 +144,28 @@ preds = model("Thank you for your email. Please go ahead and issue. Please invoi
144
  - num_epochs: (2, 2)
145
  - max_steps: -1
146
  - sampling_strategy: oversampling
147
- - num_iterations: 1
148
- - body_learning_rate: (0.0003478257195882986, 0.0003478257195882986)
149
- - head_learning_rate: 0.0003478257195882986
150
  - loss: CosineSimilarityLoss
151
  - distance_metric: cosine_distance
152
  - margin: 0.25
153
  - end_to_end: False
154
  - use_amp: False
155
  - warmup_proportion: 0.1
 
156
  - seed: 42
157
  - eval_max_steps: -1
158
- - load_best_model_at_end: False
159
 
160
  ### Training Results
161
- | Epoch | Step | Training Loss | Validation Loss |
162
- |:------:|:----:|:-------------:|:---------------:|
163
- | 0.0909 | 1 | 0.2423 | - |
 
 
164
 
 
165
  ### Framework Versions
166
  - Python: 3.10.12
167
  - SetFit: 1.1.0.dev0
 
144
  - num_epochs: (2, 2)
145
  - max_steps: -1
146
  - sampling_strategy: oversampling
147
+ - num_iterations: 2
148
+ - body_learning_rate: (0.0005201181161511404, 0.0005201181161511404)
149
+ - head_learning_rate: 0.00021200244124154418
150
  - loss: CosineSimilarityLoss
151
  - distance_metric: cosine_distance
152
  - margin: 0.25
153
  - end_to_end: False
154
  - use_amp: False
155
  - warmup_proportion: 0.1
156
+ - max_length: 512
157
  - seed: 42
158
  - eval_max_steps: -1
159
+ - load_best_model_at_end: True
160
 
161
  ### Training Results
162
+ | Epoch | Step | Training Loss | Validation Loss |
163
+ |:-------:|:------:|:-------------:|:---------------:|
164
+ | 0.0476 | 1 | 0.2504 | - |
165
+ | 1.0 | 21 | - | 0.0691 |
166
+ | **2.0** | **42** | **-** | **0.0445** |
167
 
168
+ * The bold row denotes the saved checkpoint.
169
  ### Framework Versions
170
  - Python: 3.10.12
171
  - SetFit: 1.1.0.dev0
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "sentence-transformers/all-MiniLM-L6-v2",
3
  "architectures": [
4
  "BertModel"
5
  ],
 
1
  {
2
+ "_name_or_path": "checkpoints/step_42",
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:7eace4fcc4d8b481344324ec12237f6f5c298d3185e89dbccf8af6c7ce4eda79
3
  size 90864192
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6d5bbed3f8e6e3c26eff6f9789b1143287d78fede70452eda4cef3333b833f1
3
  size 90864192
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:72d4e1bc25dfe4c123670bcacefc119d2d13c6598a487b221c85e8109d6c68b0
3
  size 44071
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ccc2b30a2521323751b38ef4e29d5ee8e2b6a71e33d25feb0e726b505e7f89dd
3
  size 44071