eunyounglee commited on
Commit
3daabd0
1 Parent(s): f4d29d4

End of training

Browse files
Files changed (1) hide show
  1. README.md +23 -4
README.md CHANGED
@@ -3,6 +3,9 @@ license: cc-by-sa-4.0
3
  base_model: klue/bert-base
4
  tags:
5
  - generated_from_trainer
 
 
 
6
  model-index:
7
  - name: pogny-persona-emotion-bert-finetuning-1
8
  results: []
@@ -14,6 +17,10 @@ should probably proofread and complete it, then remove this comment. -->
14
  # pogny-persona-emotion-bert-finetuning-1
15
 
16
  This model is a fine-tuned version of [klue/bert-base](https://huggingface.co/klue/bert-base) on the None dataset.
 
 
 
 
17
 
18
  ## Model description
19
 
@@ -42,11 +49,23 @@ The following hyperparameters were used during training:
42
 
43
  ### Training results
44
 
 
 
 
 
 
 
 
 
 
 
 
 
45
 
46
 
47
  ### Framework versions
48
 
49
- - Transformers 4.32.1
50
- - Pytorch 2.2.0+cu121
51
- - Datasets 2.12.0
52
- - Tokenizers 0.13.2
 
3
  base_model: klue/bert-base
4
  tags:
5
  - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ - f1
9
  model-index:
10
  - name: pogny-persona-emotion-bert-finetuning-1
11
  results: []
 
17
  # pogny-persona-emotion-bert-finetuning-1
18
 
19
  This model is a fine-tuned version of [klue/bert-base](https://huggingface.co/klue/bert-base) on the None dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 1.4030
22
+ - Accuracy: 0.8256
23
+ - F1: 0.8253
24
 
25
  ## Model description
26
 
 
49
 
50
  ### Training results
51
 
52
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
53
+ |:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|
54
+ | 0.487 | 1.0 | 4153 | 0.4409 | 0.8322 | 0.8276 |
55
+ | 0.374 | 2.0 | 8306 | 0.4491 | 0.8342 | 0.8330 |
56
+ | 0.2794 | 3.0 | 12459 | 0.5038 | 0.8338 | 0.8316 |
57
+ | 0.1871 | 4.0 | 16612 | 0.6531 | 0.8335 | 0.8322 |
58
+ | 0.1428 | 5.0 | 20765 | 0.8888 | 0.8258 | 0.8240 |
59
+ | 0.0968 | 6.0 | 24918 | 1.0803 | 0.8186 | 0.8186 |
60
+ | 0.076 | 7.0 | 29071 | 1.1708 | 0.8267 | 0.8261 |
61
+ | 0.0516 | 8.0 | 33224 | 1.2664 | 0.8270 | 0.8257 |
62
+ | 0.0368 | 9.0 | 37377 | 1.3662 | 0.8270 | 0.8264 |
63
+ | 0.0233 | 10.0 | 41530 | 1.4030 | 0.8256 | 0.8253 |
64
 
65
 
66
  ### Framework versions
67
 
68
+ - Transformers 4.38.1
69
+ - Pytorch 2.2.0
70
+ - Datasets 2.17.1
71
+ - Tokenizers 0.15.2