cardiffnlp/tweet_eval
Viewer • Updated • 201k • 28.7k • 147
How to use muhtasham/mini-mlm-imdb-target-tweet with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="muhtasham/mini-mlm-imdb-target-tweet") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("muhtasham/mini-mlm-imdb-target-tweet")
model = AutoModelForSequenceClassification.from_pretrained("muhtasham/mini-mlm-imdb-target-tweet", device_map="auto")This model is a fine-tuned version of muhtasham/mini-mlm-imdb on the tweet_eval 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 |
|---|---|---|---|---|---|
| 0.8543 | 4.9 | 500 | 0.6920 | 0.7674 | 0.7571 |
| 0.3797 | 9.8 | 1000 | 0.7231 | 0.7727 | 0.7709 |
| 0.1668 | 14.71 | 1500 | 0.9171 | 0.7594 | 0.7583 |
| 0.068 | 19.61 | 2000 | 1.1558 | 0.7647 | 0.7642 |
| 0.0409 | 24.51 | 2500 | 1.3042 | 0.7674 | 0.7669 |