update model card README.md
Browse files
README.md
CHANGED
@@ -11,16 +11,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
11 |
|
12 |
# GPT2-CLS-Finetuned-MBTI-GPT2-CLS-Finetuned-MBTI-JointGPT2-Warmup-from-CLS
|
13 |
|
14 |
-
This model is a fine-tuned version of [](https://huggingface.co/) on the None dataset.
|
15 |
-
It achieves the following results on the evaluation set:
|
16 |
-
- Loss: 4.4722
|
17 |
-
- Cls loss: 1.8206
|
18 |
-
- Lm loss: 4.0167
|
19 |
-
- Cls Accuracy: 0.6121
|
20 |
-
- Cls F1: 0.6063
|
21 |
-
- Cls Precision: 0.6096
|
22 |
-
- Cls Recall: 0.6121
|
23 |
-
- Perplexity: 55.52
|
24 |
|
25 |
## Model description
|
26 |
|
@@ -47,15 +38,6 @@ The following hyperparameters were used during training:
|
|
47 |
- lr_scheduler_type: linear
|
48 |
- num_epochs: 3
|
49 |
|
50 |
-
### Training results
|
51 |
-
|
52 |
-
| Training Loss | Epoch | Step | Validation Loss | Cls loss | Lm loss | Cls Accuracy | Cls F1 | Cls Precision | Cls Recall | Perplexity |
|
53 |
-
|:-------------:|:-----:|:-----:|:---------------:|:--------:|:-------:|:------------:|:------:|:-------------:|:----------:|:----------:|
|
54 |
-
| 4.5259 | 1.0 | 3470 | 4.4772 | 1.6618 | 4.0615 | 0.6063 | 0.5983 | 0.6160 | 0.6063 | 58.06 |
|
55 |
-
| 4.266 | 2.0 | 6940 | 4.4560 | 1.7186 | 4.0260 | 0.6179 | 0.6114 | 0.6210 | 0.6179 | 56.04 |
|
56 |
-
| 4.1591 | 3.0 | 10410 | 4.4722 | 1.8206 | 4.0167 | 0.6121 | 0.6063 | 0.6096 | 0.6121 | 55.52 |
|
57 |
-
|
58 |
-
|
59 |
### Framework versions
|
60 |
|
61 |
- Transformers 4.21.2
|
|
|
11 |
|
12 |
# GPT2-CLS-Finetuned-MBTI-GPT2-CLS-Finetuned-MBTI-JointGPT2-Warmup-from-CLS
|
13 |
|
14 |
+
This model is a fine-tuned version of [GItaf/GPT2-CLS-Finetuned-MBTI](https://huggingface.co/GItaf/GPT2-CLS-Finetuned-MBTI) on the None dataset.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
|
16 |
## Model description
|
17 |
|
|
|
38 |
- lr_scheduler_type: linear
|
39 |
- num_epochs: 3
|
40 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
41 |
### Framework versions
|
42 |
|
43 |
- Transformers 4.21.2
|