cardiffnlp/tweet_eval
Viewer • Updated • 201k • 28.7k • 147
How to use muhtasham/medium-mlm-imdb-target-tweet with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="muhtasham/medium-mlm-imdb-target-tweet") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("muhtasham/medium-mlm-imdb-target-tweet")
model = AutoModelForSequenceClassification.from_pretrained("muhtasham/medium-mlm-imdb-target-tweet", device_map="auto")This model is a fine-tuned version of muhtasham/medium-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.456 | 4.9 | 500 | 0.8890 | 0.7754 | 0.7720 |
| 0.0578 | 9.8 | 1000 | 1.3492 | 0.7540 | 0.7509 |
| 0.0173 | 14.71 | 1500 | 1.6143 | 0.7594 | 0.7584 |
| 0.0124 | 19.61 | 2000 | 1.6869 | 0.7620 | 0.7599 |