Views
No views yet
distilbert-base-uncased fine-tuned on a subset of the
amazon_polarity dataset for
binary POSITIVE / NEGATIVE sentiment classification of product reviews.| Model | Accuracy | F1 |
|---|---|---|
| Baseline (movie-trained, SST-2) | 88.6% | 0.888 |
| This model (product reviews) | 89.7% | 0.899 |
1from transformers import pipeline
2clf = pipeline("sentiment-analysis", model="rahiqr/distilbert-amazon-sentiment")
3clf("Arrived quickly and works perfectly!")