DistilBERT — Amazon Product Review Sentiment

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.

Results (1,000 held-out product reviews)

Model Accuracy F1
Baseline (movie-trained, SST-2) 88.6% 0.888
This model (product reviews) 89.7% 0.899

Usage

from transformers import pipeline
clf = pipeline("sentiment-analysis", model="rahiqr/distilbert-amazon-sentiment")
clf("Arrived quickly and works perfectly!")
Downloads last month
19
Safetensors
Model size
67M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for rahiqr/distilbert-amazon-sentiment

Finetuned
(12050)
this model

Dataset used to train rahiqr/distilbert-amazon-sentiment