nyu-mll/glue
Viewer • Updated • 1.49M • 470k • 497
How to use EstherT/sentence-acceptability with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="EstherT/sentence-acceptability") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("EstherT/sentence-acceptability")
model = AutoModelForSequenceClassification.from_pretrained("EstherT/sentence-acceptability")This model is a fine-tuned version of bert-base-uncased on the glue dataset. It achieves the following results on the evaluation set:
This model classifies English sentences according to two different labels: 1 if the sentence is grammatically acceptable and 0 if the sentence is grammatically unacceptable.
The model was trained on the "cola" split of the glue dataset, using the 8551 instances of its "train" split. For the evaluation, the 1043 sentences of the "evaluation" split were used.
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|---|---|---|---|
| 0.4868 | 1.0 | 1069 | 0.6279 | 0.7862 |
| 0.3037 | 2.0 | 2138 | 0.6184 | 0.8140 |
| 0.177 | 3.0 | 3207 | 0.8257 | 0.8217 |