You can deploy this model and start earning money today!

TheOnlyAtrix/
imdb-sentiment-distilbert

Name: imdb-sentiment-distilbert
Availability: InStock

safetensors

distilbert

Views

No views yet

IMDB Sentiment Classifier

Fine-tuned DistilBERT for binary sentiment classification on movie reviews.

Model Details

Base model: distilbert-base-uncased
Task: Sentiment analysis (positive / negative)
Dataset: IMDB Movie Reviews (25,000 training examples)
Training time: ~2 minutes on T4 GPU

Performance

Metric	Score
Accuracy	93.2%
F1	93.2%
Precision	93.8%
Recall	92.7%

Usage

python

from transformers import pipeline

classifier = pipeline(
    "sentiment-analysis",
    model="TheOnlyAtrix/imdb-sentiment-distilbert"
)

result = classifier("This film was an absolute masterpiece!")
print(result)
# [{'label': 'LABEL_1', 'score': 0.983}]  → positive

Labels

LABEL_0 → negative
LABEL_1 → positive

Training Details

Optimizer: AdamW
Learning rate: 2e-5
Batch size: 16
Epochs: 2
Warmup steps: 500
Weight decay: 0.01