tm21cy commited on
Commit
fecc01b
·
verified ·
1 Parent(s): f06f092

End of training

Browse files
README.md CHANGED
@@ -32,7 +32,7 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the emotion dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.1786
36
  - Accuracy: 0.929
37
 
38
  ## Model description
@@ -52,21 +52,22 @@ More information needed
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
- - learning_rate: 5e-05
56
  - train_batch_size: 16
57
  - eval_batch_size: 16
58
  - seed: 42
59
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
  - lr_scheduler_type: linear
61
- - num_epochs: 3
62
 
63
  ### Training results
64
 
65
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
- | 0.218 | 1.0 | 1000 | 0.1718 | 0.9335 |
68
- | 0.1272 | 2.0 | 2000 | 0.1489 | 0.9345 |
69
- | 0.0924 | 3.0 | 3000 | 0.1632 | 0.9435 |
 
70
 
71
 
72
  ### Framework versions
 
32
 
33
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the emotion dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.3674
36
  - Accuracy: 0.929
37
 
38
  ## Model description
 
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
+ - learning_rate: 8.42e-05
56
  - train_batch_size: 16
57
  - eval_batch_size: 16
58
  - seed: 42
59
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
  - lr_scheduler_type: linear
61
+ - num_epochs: 4
62
 
63
  ### Training results
64
 
65
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
+ | 0.1807 | 1.0 | 1000 | 0.2292 | 0.927 |
68
+ | 0.1326 | 2.0 | 2000 | 0.2684 | 0.9305 |
69
+ | 0.0675 | 3.0 | 3000 | 0.2977 | 0.938 |
70
+ | 0.0341 | 4.0 | 4000 | 0.3249 | 0.9385 |
71
 
72
 
73
  ### Framework versions
runs/Mar11_06-21-33_52dd5b86094b/events.out.tfevents.1710138094.52dd5b86094b.7559.4 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c48a0a0363e28eec6eab79dd6a9680f9b561472d9074160302aa699587836346
3
- size 7649
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0c9670558cace588b62613a86031de3215a07d6b4b06345d1f25aac3092f470
3
+ size 8326
runs/Mar11_06-21-33_52dd5b86094b/events.out.tfevents.1710140119.52dd5b86094b.7559.5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ab85fd43cf5ce1ba98817d06126ec446e5b6da3d675fbad14c8450cceaf3bf0
3
+ size 411
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c5cc3370f5bab2ffcc00822dc9a2fbcb22e9cfea1d249ad0bc149559220e61bc
3
  size 4920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f2ef10befda6602b452412286e0cf970bcad751d3942c8ddf2e5a3c544cd2c9
3
  size 4920