Views
No views yet
distilbert-base-uncased on the IMDB movie review dataset for binary sentiment classification (positive / negative).| Metric | Score |
|---|---|
| Test accuracy | 91.33% |
| Training loss | 0.238 |
| Validation loss | 0.225 |
| Epochs | 1 |
1from transformers import pipeline
2
3classifier = pipeline(
4 "text-classification",
5 model="BlessedDovis/distilbert-sentiment-analysis"
6)
7
8result = classifier("This movie was absolutely fantastic!")
9# POSITIVE — 99.52% confidence