stanfordnlp/imdb
Viewer • Updated • 100k • 174k • 472
How to use amanda-cristina/my_awesome_model with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="amanda-cristina/my_awesome_model") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("amanda-cristina/my_awesome_model")
model = AutoModelForSequenceClassification.from_pretrained("amanda-cristina/my_awesome_model", device_map="auto")This model is a fine-tuned version of kiddothe2b/longformer-mini-1024 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 | F1 |
|---|---|---|---|---|---|
| No log | 1.0 | 375 | 0.5384 | 0.7367 | 0.7492 |
| 0.5846 | 2.0 | 750 | 0.4475 | 0.79 | 0.7907 |