Hugging Face's logo

PavanNeerudu
/
gpt2-finetuned-sst2

Text Classification
Transformers
PyTorch
English
gpt2
Eval Results
text-generation-inference
Model card Files Community
1
PavanNeerudu commited on
Commit
a8e2c96
·
1 Parent(s): a157b18

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -6
README.md CHANGED
@@ -19,7 +19,7 @@ model-index:
19
  metrics:
20
  - name: Accuracy
21
  type: accuracy
22
- value: 0.92775
23
  ---
24
 
25
 
@@ -28,7 +28,7 @@ model-index:
28
  <!-- Provide a quick summary of what the model is/does. -->
29
 
30
  This model is GPT-2 fine-tuned on GLUE SST-2 dataset. It acheives the following results on the validation set
31
- - Accuracy: 0.92775
32
 
33
 
34
  ## Model Details
@@ -54,7 +54,6 @@ The following hyperparameters were used during training:
54
 
55
  |Epoch | Training Loss | Training Accuracy | Validation Loss | Validation Accuracy |
56
  |:----:|:-------------:|:-----------------:|:---------------:|:-------------------:|
57
- | 1 | 0.28664 | 0.87905 | 0.25106 | 0.91514 |
58
- | 2 | 0.17448 | 0.93327 | 0.27479 | 0.91399 |
59
- | 3 | 0.13553 | 0.95191 | 0.30146 | 0.90826 |
60
- | 4 | 0.11361 | 0.96059 | 0.29815 | **0.92775** |
 
19
  metrics:
20
  - name: Accuracy
21
  type: accuracy
22
+ value: 0.9254
23
  ---
24
 
25
 
 
28
  <!-- Provide a quick summary of what the model is/does. -->
29
 
30
  This model is GPT-2 fine-tuned on GLUE SST-2 dataset. It acheives the following results on the validation set
31
+ - Accuracy: 0.9254
32
 
33
 
34
  ## Model Details
 
54
 
55
  |Epoch | Training Loss | Training Accuracy | Validation Loss | Validation Accuracy |
56
  |:----:|:-------------:|:-----------------:|:---------------:|:-------------------:|
57
+ | 1 | 0.32641 | 0.85419 | 0.26545 | 0.90137 |
58
+ | 2 | 0.15731 | 0.94151 | 0.23625 | **0.92546** |
59
+ | 3 | 0.08982 | 0.9712 | 0.33954 | 0.91514 |