fancyzhx/amazon_polarity
Viewer • Updated • 4M • 20.2k • 51
distilbert-base-uncased fine-tuned on a subset of the
amazon_polarity dataset for
binary POSITIVE / NEGATIVE sentiment classification of product reviews.
Motivation: the common off-the-shelf sentiment model is trained on movie reviews (SST-2). This model adapts to the product-review domain.
| Model | Accuracy | F1 |
|---|---|---|
| Baseline (movie-trained, SST-2) | 88.6% | 0.888 |
| This model (product reviews) | 89.7% | 0.899 |
from transformers import pipeline
clf = pipeline("sentiment-analysis", model="rahiqr/distilbert-amazon-sentiment")
clf("Arrived quickly and works perfectly!")
Base model
distilbert/distilbert-base-uncased