nyu-mll/glue
Viewer • Updated • 1.49M • 685k • 775
How to use gokulsrinivasagan/bert_base_train_qqp with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="gokulsrinivasagan/bert_base_train_qqp") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("gokulsrinivasagan/bert_base_train_qqp")
model = AutoModelForSequenceClassification.from_pretrained("gokulsrinivasagan/bert_base_train_qqp", device_map="auto")This model is a fine-tuned version of gokulsrinivasagan/bert_base_train on the GLUE QQP 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 | Accuracy | F1 | Combined Score |
|---|---|---|---|---|---|---|
| 0.4435 | 1.0 | 1422 | 0.3833 | 0.8191 | 0.7372 | 0.7781 |
| 0.3251 | 2.0 | 2844 | 0.3589 | 0.8375 | 0.7846 | 0.8110 |
| 0.244 | 3.0 | 4266 | 0.4182 | 0.8469 | 0.7928 | 0.8199 |
| 0.1843 | 4.0 | 5688 | 0.4489 | 0.8504 | 0.7882 | 0.8193 |
| 0.1434 | 5.0 | 7110 | 0.4336 | 0.8516 | 0.7966 | 0.8241 |
| 0.1145 | 6.0 | 8532 | 0.4963 | 0.8507 | 0.7904 | 0.8205 |
| 0.0939 | 7.0 | 9954 | 0.5458 | 0.8491 | 0.7990 | 0.8241 |
Base model
distilbert/distilbert-base-uncased