stanfordnlp/imdb
Viewer • Updated • 100k • 205k • 374
How to use ketong3906/my_awesome_model_classification_w_adapter with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="ketong3906/my_awesome_model_classification_w_adapter") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("ketong3906/my_awesome_model_classification_w_adapter")
model = AutoModelForSequenceClassification.from_pretrained("ketong3906/my_awesome_model_classification_w_adapter")This model is a fine-tuned version of roberta-base on the imdb 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 |
|---|---|---|---|---|
| No log | 1.0 | 15 | 0.0120 | 1.0 |
| No log | 2.0 | 30 | 0.0035 | 1.0 |