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
8b16318
•
1 Parent(s): 5224a0a

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +5 -8
  2. model.safetensors +1 -1
  3. 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: (16, 16)
144
  - num_epochs: (2, 2)
145
  - max_steps: -1
146
  - sampling_strategy: oversampling
147
- - num_iterations: 2
148
- - body_learning_rate: (0.0002853593106019776, 0.0002853593106019776)
149
- - head_learning_rate: 0.0002853593106019776
150
  - loss: CosineSimilarityLoss
151
  - distance_metric: cosine_distance
152
  - margin: 0.25
@@ -160,10 +160,7 @@ preds = model("Thank you for your email. Please go ahead and issue. Please invoi
160
  ### Training Results
161
  | Epoch | Step | Training Loss | Validation Loss |
162
  |:------:|:----:|:-------------:|:---------------:|
163
- | 0.0119 | 1 | 0.2534 | - |
164
- | 0.5952 | 50 | 0.0849 | - |
165
- | 1.1905 | 100 | 0.0077 | - |
166
- | 1.7857 | 150 | 0.0099 | - |
167
 
168
  ### Framework Versions
169
  - Python: 3.10.12
 
140
  | 13 | 24 |
141
 
142
  ### Training Hyperparameters
143
+ - batch_size: (64, 64)
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
 
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
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c7ea3d24a2a63f6ea05412255c9f1061e05d36e615e70f248f15c5335af11654
3
  size 90864192
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7eace4fcc4d8b481344324ec12237f6f5c298d3185e89dbccf8af6c7ce4eda79
3
  size 90864192
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:23618d3911cbfd70dee3ae87baf7516bd1d76add50e410389e6f27aade672c16
3
  size 44071
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72d4e1bc25dfe4c123670bcacefc119d2d13c6598a487b221c85e8109d6c68b0
3
  size 44071