PavanNeerudu commited on
Commit
1116bde
·
1 Parent(s): 591328e

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +58 -0
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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-cola
11
+ results:
12
+ - task:
13
+ name: Text Classification
14
+ type: text-classification
15
+ dataset:
16
+ name: GLUE COLA
17
+ type: glue
18
+ args: cola
19
+ metrics:
20
+ - name: Accuracy
21
+ type: accuracy
22
+ value: 0.77756
23
+ ---
24
+
25
+
26
+ # gpt2-finetuned-cola
27
+
28
+ <!-- Provide a quick summary of what the model is/does. -->
29
+
30
+ This model is gpt2 fine-tuned on GLUE CoLA dataset. It acheives the following results on the validation set. Though MCC is used for CoLA dataset, accuracy is reported here
31
+ - Accuracy: 0.77756
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.
37
+ More precisely, it was trained to guess the next word in sentences. However, it acheives very good results on Text Classification tasks.
38
+
39
+ ## Training procedure
40
+
41
+ ### Training hyperparameters
42
+
43
+ The following hyperparameters were used during training:
44
+ - learning_rate: 3e-4
45
+ - train_batch_size: 32
46
+ - eval_batch_size: 32
47
+ - seed: 42
48
+ - optimizer: epsilon=1e-08
49
+ - num_epochs: 5.0
50
+
51
+ ### Training results
52
+
53
+ |Epoch | Training Loss | Validation Loss | Training Accuracy | Validation Accuracy |
54
+ | 1 | 0.65824 | 0.55883 | 0.67957 | 0.70086 |
55
+ | 2 | 0.51774 | 0.53402 | 0.74927 | 0.75264 |
56
+ | 3 | 0.44502 | 0.50641 | 0.79172 | **0.77756** |
57
+ | 4 | 0.38996 | 0.54241 | 0.82645 | 0.77469 |
58
+ | 5 | 0.35218 | 0.55088 | 0.84516 | 0.77277 |