GItaf commited on
Commit
301c402
1 Parent(s): 5b0b191

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -1
README.md CHANGED
@@ -11,7 +11,16 @@ 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 [GItaf/GPT2-CLS-Finetuned-MBTI](https://huggingface.co/GItaf/GPT2-CLS-Finetuned-MBTI) on the None dataset.
 
 
 
 
 
 
 
 
 
15
 
16
  ## Model description
17
 
@@ -38,6 +47,15 @@ The following hyperparameters were used during training:
38
  - lr_scheduler_type: linear
39
  - num_epochs: 3
40
 
 
 
 
 
 
 
 
 
 
41
  ### Framework versions
42
 
43
  - 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 [](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
  - 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