Commit
·
05bb647
1
Parent(s):
da04a09
Create README.md
Browse files
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** |
|