PavanNeerudu commited on
Commit
05bb647
·
1 Parent(s): da04a09

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-qnli
11
+ results:
12
+ - task:
13
+ name: Text Classification
14
+ type: text-classification
15
+ dataset:
16
+ name: GLUE QNLI
17
+ type: glue
18
+ args: qnli
19
+ metrics:
20
+ - name: Accuracy
21
+ type: accuracy
22
+ value: 0.87791
23
+ ---
24
+
25
+
26
+ # gpt2-finetuned-qnli
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.87791
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: 8
46
+ - eval_batch_size: 8
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.47803 | 0.77344 | 0.34630 | 0.85502 |
58
+ | 2 | 0.36961 | 0.85486 | 0.32994 | 0.85502 |
59
+ | 3 | 0.32343 | 0.88785 | 0.38374 | **0.87791** |