PavanNeerudu commited on
Commit
02a9166
·
1 Parent(s): 4c5a6fa

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +59 -0
README.md ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ license: apache-2.0
5
+ datasets:
6
+ - glue
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: gpt2-finetuned-wnli
11
+ results:
12
+ - task:
13
+ name: Text Classification
14
+ type: text-classification
15
+ dataset:
16
+ name: GLUE WNLI
17
+ type: glue
18
+ args: wnli
19
+ metrics:
20
+ - name: Accuracy
21
+ type: accuracy
22
+ value: 0.54930
23
+ ---
24
+
25
+
26
+ # gpt2-finetuned-wnli
27
+
28
+ <!-- Provide a quick summary of what the model is/does. -->
29
+
30
+ This model is GPT-2 fine-tuned on GLUE STS-B dataset. It acheives the following results on the validation set
31
+ - Accuracy: 0.54930
32
+
33
+
34
+ ## Model Details
35
+ GPT-2 is a transformers model pretrained on a very large corpus of English data in a self-supervised fashion.
36
+ This means it was pretrained on the raw texts only, with no humans labelling them in any way (which is why it can use lots of publicly available data) with an automatic process to generate inputs and labels from those texts. More precisely, it was trained to guess the next word in sentences.
37
+ However, it acheives very good results on Text Classification tasks.
38
+ ## Training procedure
39
+
40
+
41
+ ### Training hyperparameters
42
+
43
+ The following hyperparameters were used during training:
44
+ - learning_rate: 2e-5
45
+ - train_batch_size: 16
46
+ - eval_batch_size: 16
47
+ - seed: 123
48
+ - optimizer: epsilon=1e-08
49
+ - num_epochs: 3
50
+
51
+ ### Training results
52
+
53
+
54
+
55
+ |Epoch | Training Loss | Training Accuracy | Validation Loss | Validation Accuracy |
56
+ |:----:|:-------------:|:-----------------:|:---------------:|:-------------------:|
57
+ | 1 | 0.72133 | 0.49449 | 0.67626 | 0.50704 |
58
+ | 2 | 0.71982 | 0.50866 | 0.70278 | 0.49296 |
59
+ | 3 | 0.70411 | 0.51181 | 0.68919 | **0.54930** |