henryscheible commited on
Commit
301a4ae
1 Parent(s): ec802f1

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +79 -2
README.md CHANGED
@@ -2,9 +2,26 @@
2
  license: mit
3
  tags:
4
  - generated_from_trainer
 
 
 
 
5
  model-index:
6
  - name: gpt2_crows_pairs_finetuned
7
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  ---
9
 
10
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -12,7 +29,14 @@ should probably proofread and complete it, then remove this comment. -->
12
 
13
  # gpt2_crows_pairs_finetuned
14
 
15
- This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
 
 
 
 
 
 
 
16
 
17
  ## Model description
18
 
@@ -39,6 +63,59 @@ The following hyperparameters were used during training:
39
  - lr_scheduler_type: linear
40
  - num_epochs: 50
41
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
42
  ### Framework versions
43
 
44
  - Transformers 4.26.1
 
2
  license: mit
3
  tags:
4
  - generated_from_trainer
5
+ datasets:
6
+ - crows_pairs
7
+ metrics:
8
+ - accuracy
9
  model-index:
10
  - name: gpt2_crows_pairs_finetuned
11
+ results:
12
+ - task:
13
+ name: Text Classification
14
+ type: text-classification
15
+ dataset:
16
+ name: crows_pairs
17
+ type: crows_pairs
18
+ config: crows_pairs
19
+ split: test
20
+ args: crows_pairs
21
+ metrics:
22
+ - name: Accuracy
23
+ type: accuracy
24
+ value: 0.5066225165562914
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
29
 
30
  # gpt2_crows_pairs_finetuned
31
 
32
+ This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the crows_pairs dataset.
33
+ It achieves the following results on the evaluation set:
34
+ - Loss: 5.0353
35
+ - Accuracy: 0.5066
36
+ - Tp: 0.3046
37
+ - Tn: 0.2020
38
+ - Fp: 0.2748
39
+ - Fn: 0.2185
40
 
41
  ## Model description
42
 
 
63
  - lr_scheduler_type: linear
64
  - num_epochs: 50
65
 
66
+ ### Training results
67
+
68
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Tp | Tn | Fp | Fn |
69
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:------:|:------:|:------:|
70
+ | 0.9942 | 1.05 | 20 | 0.6944 | 0.4801 | 0.0464 | 0.4338 | 0.0430 | 0.4768 |
71
+ | 0.6962 | 2.11 | 40 | 0.6909 | 0.5397 | 0.4205 | 0.1192 | 0.3576 | 0.1026 |
72
+ | 0.6711 | 3.16 | 60 | 0.7195 | 0.5 | 0.3344 | 0.1656 | 0.3113 | 0.1887 |
73
+ | 0.5148 | 4.21 | 80 | 0.8547 | 0.5232 | 0.2583 | 0.2649 | 0.2119 | 0.2649 |
74
+ | 0.3053 | 5.26 | 100 | 1.1929 | 0.4868 | 0.2483 | 0.2384 | 0.2384 | 0.2748 |
75
+ | 0.1268 | 6.32 | 120 | 1.7763 | 0.4901 | 0.2384 | 0.2517 | 0.2252 | 0.2848 |
76
+ | 0.0999 | 7.37 | 140 | 2.3687 | 0.5066 | 0.2914 | 0.2152 | 0.2616 | 0.2318 |
77
+ | 0.0374 | 8.42 | 160 | 2.3493 | 0.5099 | 0.2781 | 0.2318 | 0.2450 | 0.2450 |
78
+ | 0.009 | 9.47 | 180 | 5.1845 | 0.4868 | 0.3179 | 0.1689 | 0.3079 | 0.2053 |
79
+ | 0.0316 | 10.53 | 200 | 3.6149 | 0.4934 | 0.3013 | 0.1921 | 0.2848 | 0.2219 |
80
+ | 0.0223 | 11.58 | 220 | 3.7626 | 0.4934 | 0.2384 | 0.2550 | 0.2219 | 0.2848 |
81
+ | 0.0295 | 12.63 | 240 | 2.9512 | 0.5232 | 0.3411 | 0.1821 | 0.2947 | 0.1821 |
82
+ | 0.0248 | 13.68 | 260 | 3.5709 | 0.5066 | 0.3146 | 0.1921 | 0.2848 | 0.2086 |
83
+ | 0.0387 | 14.74 | 280 | 3.5248 | 0.4801 | 0.2318 | 0.2483 | 0.2285 | 0.2914 |
84
+ | 0.0223 | 15.79 | 300 | 3.7220 | 0.5066 | 0.2483 | 0.2583 | 0.2185 | 0.2748 |
85
+ | 0.0223 | 16.84 | 320 | 4.1806 | 0.5033 | 0.2384 | 0.2649 | 0.2119 | 0.2848 |
86
+ | 0.0254 | 17.89 | 340 | 4.4477 | 0.4934 | 0.2351 | 0.2583 | 0.2185 | 0.2881 |
87
+ | 0.0002 | 18.95 | 360 | 4.2539 | 0.5298 | 0.2649 | 0.2649 | 0.2119 | 0.2583 |
88
+ | 0.0238 | 20.0 | 380 | 3.9944 | 0.5033 | 0.2318 | 0.2715 | 0.2053 | 0.2914 |
89
+ | 0.0027 | 21.05 | 400 | 4.6360 | 0.5066 | 0.2285 | 0.2781 | 0.1987 | 0.2947 |
90
+ | 0.0001 | 22.11 | 420 | 5.0025 | 0.5132 | 0.2616 | 0.2517 | 0.2252 | 0.2616 |
91
+ | 0.0135 | 23.16 | 440 | 4.2408 | 0.5 | 0.2152 | 0.2848 | 0.1921 | 0.3079 |
92
+ | 0.0004 | 24.21 | 460 | 4.9804 | 0.5066 | 0.2781 | 0.2285 | 0.2483 | 0.2450 |
93
+ | 0.0213 | 25.26 | 480 | 5.5604 | 0.4967 | 0.3046 | 0.1921 | 0.2848 | 0.2185 |
94
+ | 0.0147 | 26.32 | 500 | 5.0523 | 0.5066 | 0.2815 | 0.2252 | 0.2517 | 0.2417 |
95
+ | 0.011 | 27.37 | 520 | 4.8651 | 0.5132 | 0.2682 | 0.2450 | 0.2318 | 0.2550 |
96
+ | 0.012 | 28.42 | 540 | 4.5382 | 0.5232 | 0.3278 | 0.1954 | 0.2815 | 0.1954 |
97
+ | 0.0024 | 29.47 | 560 | 4.0583 | 0.5033 | 0.2848 | 0.2185 | 0.2583 | 0.2384 |
98
+ | 0.0001 | 30.53 | 580 | 4.4274 | 0.4967 | 0.2980 | 0.1987 | 0.2781 | 0.2252 |
99
+ | 0.0005 | 31.58 | 600 | 4.6131 | 0.5099 | 0.2980 | 0.2119 | 0.2649 | 0.2252 |
100
+ | 0.0008 | 32.63 | 620 | 4.7104 | 0.5132 | 0.2748 | 0.2384 | 0.2384 | 0.2483 |
101
+ | 0.0 | 33.68 | 640 | 4.7753 | 0.5066 | 0.2947 | 0.2119 | 0.2649 | 0.2285 |
102
+ | 0.0 | 34.74 | 660 | 4.9226 | 0.5199 | 0.2682 | 0.2517 | 0.2252 | 0.2550 |
103
+ | 0.0 | 35.79 | 680 | 5.0117 | 0.4967 | 0.3179 | 0.1788 | 0.2980 | 0.2053 |
104
+ | 0.0 | 36.84 | 700 | 5.0534 | 0.5033 | 0.3079 | 0.1954 | 0.2815 | 0.2152 |
105
+ | 0.0 | 37.89 | 720 | 5.0638 | 0.5166 | 0.2682 | 0.2483 | 0.2285 | 0.2550 |
106
+ | 0.0 | 38.95 | 740 | 5.1010 | 0.5132 | 0.2649 | 0.2483 | 0.2285 | 0.2583 |
107
+ | 0.0 | 40.0 | 760 | 5.1367 | 0.5132 | 0.2649 | 0.2483 | 0.2285 | 0.2583 |
108
+ | 0.0001 | 41.05 | 780 | 5.1730 | 0.5166 | 0.2781 | 0.2384 | 0.2384 | 0.2450 |
109
+ | 0.0 | 42.11 | 800 | 5.0295 | 0.5232 | 0.2848 | 0.2384 | 0.2384 | 0.2384 |
110
+ | 0.0 | 43.16 | 820 | 5.0261 | 0.5166 | 0.2781 | 0.2384 | 0.2384 | 0.2450 |
111
+ | 0.0 | 44.21 | 840 | 5.0447 | 0.5166 | 0.2781 | 0.2384 | 0.2384 | 0.2450 |
112
+ | 0.0126 | 45.26 | 860 | 5.0130 | 0.5232 | 0.2980 | 0.2252 | 0.2517 | 0.2252 |
113
+ | 0.0 | 46.32 | 880 | 5.0184 | 0.5066 | 0.3046 | 0.2020 | 0.2748 | 0.2185 |
114
+ | 0.0 | 47.37 | 900 | 5.0276 | 0.5066 | 0.3046 | 0.2020 | 0.2748 | 0.2185 |
115
+ | 0.0 | 48.42 | 920 | 5.0332 | 0.5066 | 0.3046 | 0.2020 | 0.2748 | 0.2185 |
116
+ | 0.0 | 49.47 | 940 | 5.0353 | 0.5066 | 0.3046 | 0.2020 | 0.2748 | 0.2185 |
117
+
118
+
119
  ### Framework versions
120
 
121
  - Transformers 4.26.1