Downloads · 30 days
10
24% of all-time downloads
rahiqr/distilbert-amazon-sentiment
distilbert-amazon-sentiment is a text classification model from rahiqr. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
distilbert-base-uncased fine-tuned on a subset of the amazonpolarity dataset for binary POSITIVE / NEGATIVE sentiment classification of product reviews.
Downloads · 30 days
10
24% of all-time downloads
All-time downloads
41
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors268 MB · 100%
From the Hugging Face model README
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!")