Downloads · 30 days
10
0% of all-time downloads
PavanNeerudu/gpt2-finetuned-cola
gpt2-finetuned-cola is a text classification model from PavanNeerudu. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
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 - Accuracy: 0.77756
Downloads · 30 days
10
0% of all-time downloads
All-time downloads
6.1K
Public
Repo size
1 GB
Likes
2
Public
Click a slice to open those files.
.bin510 MB · 100%
From the Hugging Face model README
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
GPT-2 is a transformers model pretrained on a very large corpus of English data in a self-supervised fashion. 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. However, it acheives very good results on Text Classification tasks.
The following hyperparameters were used during training:
| Epoch | Training Loss | Validation Loss | Training Accuracy | Validation Accuracy |
|---|---|---|---|---|
| 1 | 0.65824 | 0.55883 | 0.67957 | 0.70086 |
| 2 | 0.51774 | 0.53402 | 0.74927 | 0.75264 |
| 3 | 0.44502 | 0.50641 | 0.79172 | 0.77756 |
| 4 | 0.38996 | 0.54241 | 0.82645 | 0.77469 |
| 5 | 0.35218 | 0.55088 | 0.84516 | 0.77277 |