Mustafa-AbdulRazek commited on
Commit
cfedb22
1 Parent(s): 22d3ecd

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -6
README.md CHANGED
@@ -16,9 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the emotion dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.2153
20
- - Acc : 0.9235
21
- - F1 : 0.9234
22
 
23
  ## Model description
24
 
@@ -43,14 +43,22 @@ The following hyperparameters were used during training:
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
- - num_epochs: 2
47
 
48
  ### Training results
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Acc | F1 |
51
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|
52
- | 0.8475 | 1.0 | 250 | 0.3152 | 0.904 | 0.9014 |
53
- | 0.2462 | 2.0 | 500 | 0.2153 | 0.9235 | 0.9234 |
 
 
 
 
 
 
 
 
54
 
55
 
56
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the emotion dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.2433
20
+ - Acc : 0.9355
21
+ - F1 : 0.9356
22
 
23
  ## Model description
24
 
 
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
+ - num_epochs: 10
47
 
48
  ### Training results
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Acc | F1 |
51
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|
52
+ | 0.1674 | 1.0 | 250 | 0.1948 | 0.9275 | 0.9267 |
53
+ | 0.1185 | 2.0 | 500 | 0.1635 | 0.938 | 0.9380 |
54
+ | 0.0998 | 3.0 | 750 | 0.1723 | 0.9345 | 0.9352 |
55
+ | 0.0808 | 4.0 | 1000 | 0.1687 | 0.934 | 0.9337 |
56
+ | 0.0621 | 5.0 | 1250 | 0.1769 | 0.937 | 0.9368 |
57
+ | 0.0511 | 6.0 | 1500 | 0.1927 | 0.933 | 0.9327 |
58
+ | 0.0393 | 7.0 | 1750 | 0.2275 | 0.9345 | 0.9348 |
59
+ | 0.0323 | 8.0 | 2000 | 0.2338 | 0.932 | 0.9324 |
60
+ | 0.0273 | 9.0 | 2250 | 0.2439 | 0.935 | 0.9352 |
61
+ | 0.0215 | 10.0 | 2500 | 0.2433 | 0.9355 | 0.9356 |
62
 
63
 
64
  ### Framework versions