eunyounglee
commited on
Commit
•
3daabd0
1
Parent(s):
f4d29d4
End of training
Browse files
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.
|
50 |
-
- Pytorch 2.2.0
|
51 |
-
- Datasets 2.
|
52 |
-
- Tokenizers 0.
|
|
|
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
|