tm21cy commited on
Commit
b44c1df
·
verified ·
1 Parent(s): 04b0625

End of training

Browse files
README.md CHANGED
@@ -22,7 +22,7 @@ model-index:
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
- value: 0.927
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the emotion dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.1888
36
- - Accuracy: 0.927
37
 
38
  ## Model description
39
 
@@ -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.2348 | 1.0 | 1000 | 0.2163 | 0.9255 |
68
- | 0.1337 | 2.0 | 2000 | 0.1446 | 0.94 |
69
- | 0.0886 | 3.0 | 3000 | 0.1541 | 0.9425 |
 
70
 
71
 
72
  ### Framework versions
 
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
+ value: 0.919
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
32
 
33
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the emotion dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.4055
36
+ - Accuracy: 0.919
37
 
38
  ## Model description
39
 
 
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.1735 | 1.0 | 1000 | 0.2957 | 0.919 |
68
+ | 0.1154 | 2.0 | 2000 | 0.2450 | 0.9305 |
69
+ | 0.072 | 3.0 | 3000 | 0.2936 | 0.935 |
70
+ | 0.032 | 4.0 | 4000 | 0.3145 | 0.938 |
71
 
72
 
73
  ### Framework versions
runs/Mar11_04-54-09_f6b7ca2de819/events.out.tfevents.1710132850.f6b7ca2de819.11996.4 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0268edb2b7993b6700723d87297457f1961731dc728687d96424796797a94989
3
- size 7555
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da39fae515cf004d25816f856834e1a513d5ea11cecea00b068a046bf8fb6bb4
3
+ size 8232
runs/Mar11_04-54-09_f6b7ca2de819/events.out.tfevents.1710133918.f6b7ca2de819.11996.5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:848388080bcce3ddf652dfdb7278d5f8e79ae85c10fab363b5ff97d94dc1143c
3
+ size 411
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:75b232ae5b88203fffec359819dd835494a78f15470739cd8dade60135b72c99
3
  size 4920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0acfda34f5c210b7a98ed255121422bfae7eebc6c0740b69b494a866d43d0b24
3
  size 4920