nyu-mll/glue
Viewer โข Updated โข 1.49M โข 662k โข 783
How to use gokulsrinivasagan/bert_base_train_book_ent_2_inv_cola with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="gokulsrinivasagan/bert_base_train_book_ent_2_inv_cola") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("gokulsrinivasagan/bert_base_train_book_ent_2_inv_cola")
model = AutoModelForSequenceClassification.from_pretrained("gokulsrinivasagan/bert_base_train_book_ent_2_inv_cola", device_map="auto")This model is a fine-tuned version of gokulsrinivasagan/bert_base_train_book_ent_2_inv on the GLUE COLA dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Matthews Correlation | Accuracy |
|---|---|---|---|---|---|
| 0.6167 | 1.0 | 34 | 0.6231 | 0.0 | 0.6913 |
| 0.6124 | 2.0 | 68 | 0.6176 | 0.0 | 0.6913 |
| 0.6042 | 3.0 | 102 | 0.6221 | -0.0079 | 0.6894 |
| 0.5902 | 4.0 | 136 | 0.6327 | -0.0112 | 0.6874 |
| 0.5581 | 5.0 | 170 | 0.6435 | 0.0687 | 0.6874 |
| 0.5256 | 6.0 | 204 | 0.6595 | 0.0700 | 0.6146 |
| 0.5127 | 7.0 | 238 | 0.6980 | 0.1002 | 0.5570 |
Base model
distilbert/distilbert-base-uncased