Hugging Face's logo

gArthur98
/
Greg-Sentiment-classifier

Text Classification
Transformers
PyTorch
TensorBoard
Safetensors
bert
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
3
gArthur98 commited on
Commit
912c5ce
•
1 Parent(s): 7ca9732

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -9
README.md CHANGED
@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [microsoft/MiniLM-L12-H384-uncased](https://huggingface.co/microsoft/MiniLM-L12-H384-uncased) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 1.0165
20
  - F1: 0.3222
21
 
22
  ## Model description
@@ -36,24 +36,30 @@ More information needed
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
- - learning_rate: 0.004
40
  - train_batch_size: 16
41
  - eval_batch_size: 16
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
- - num_epochs: 6
46
 
47
  ### Training results
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | F1 |
50
  |:-------------:|:-----:|:----:|:---------------:|:------:|
51
- | 1.0386 | 1.0 | 499 | 1.0199 | 0.2347 |
52
- | 1.0437 | 2.0 | 998 | 1.0226 | 0.2347 |
53
- | 1.066 | 3.0 | 1497 | 1.0207 | 0.2347 |
54
- | 1.0183 | 4.0 | 1996 | 1.0173 | 0.3222 |
55
- | 0.9965 | 5.0 | 2495 | 1.0194 | 0.2347 |
56
- | 0.9548 | 6.0 | 2994 | 1.0165 | 0.3222 |
 
 
 
 
 
 
57
 
58
 
59
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [microsoft/MiniLM-L12-H384-uncased](https://huggingface.co/microsoft/MiniLM-L12-H384-uncased) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 1.0161
20
  - F1: 0.3222
21
 
22
  ## Model description
 
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
+ - learning_rate: 0.0005
40
  - train_batch_size: 16
41
  - eval_batch_size: 16
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
+ - num_epochs: 12
46
 
47
  ### Training results
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | F1 |
50
  |:-------------:|:-----:|:----:|:---------------:|:------:|
51
+ | 0.9805 | 1.0 | 499 | 1.0211 | 0.2347 |
52
+ | 1.0289 | 2.0 | 998 | 1.0175 | 0.2347 |
53
+ | 0.9924 | 3.0 | 1497 | 1.0189 | 0.2347 |
54
+ | 1.0319 | 4.0 | 1996 | 1.0165 | 0.3222 |
55
+ | 1.0351 | 5.0 | 2495 | 1.0179 | 0.3222 |
56
+ | 1.0339 | 6.0 | 2994 | 1.0172 | 0.3222 |
57
+ | 1.0039 | 7.0 | 3493 | 1.0163 | 0.3222 |
58
+ | 1.0299 | 8.0 | 3992 | 1.0164 | 0.3222 |
59
+ | 0.9914 | 9.0 | 4491 | 1.0168 | 0.3222 |
60
+ | 1.0038 | 10.0 | 4990 | 1.0162 | 0.3222 |
61
+ | 1.0399 | 11.0 | 5489 | 1.0161 | 0.3222 |
62
+ | 0.996 | 12.0 | 5988 | 1.0161 | 0.3222 |
63
 
64
 
65
  ### Framework versions