MRAIRR commited on
Commit
b61fc03
1 Parent(s): 645f4d2

End of training

Browse files
Files changed (1) hide show
  1. README.md +8 -6
README.md CHANGED
@@ -4,6 +4,7 @@ tags:
4
  - generated_from_trainer
5
  metrics:
6
  - accuracy
 
7
  model-index:
8
  - name: emotion_classification_model
9
  results: []
@@ -16,8 +17,9 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [klue/roberta-base](https://huggingface.co/klue/roberta-base) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.3536
20
- - Accuracy: 0.8942
 
21
 
22
  ## Model description
23
 
@@ -46,10 +48,10 @@ The following hyperparameters were used during training:
46
 
47
  ### Training results
48
 
49
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
- | 0.4965 | 1.0 | 1829 | 0.4160 | 0.8543 |
52
- | 0.2357 | 2.0 | 3658 | 0.3536 | 0.8942 |
53
 
54
 
55
  ### Framework versions
 
4
  - generated_from_trainer
5
  metrics:
6
  - accuracy
7
+ - f1
8
  model-index:
9
  - name: emotion_classification_model
10
  results: []
 
17
 
18
  This model is a fine-tuned version of [klue/roberta-base](https://huggingface.co/klue/roberta-base) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.2954
21
+ - Accuracy: 0.9079
22
+ - F1: 0.9074
23
 
24
  ## Model description
25
 
 
48
 
49
  ### Training results
50
 
51
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
52
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
53
+ | 0.4771 | 1.0 | 1829 | 0.3789 | 0.8669 | 0.8650 |
54
+ | 0.2378 | 2.0 | 3658 | 0.2954 | 0.9079 | 0.9074 |
55
 
56
 
57
  ### Framework versions