DistilBERT IMDb Sentiment Classifier
Fine-tuned distilbert-base-uncased for binary sentiment classification.
Model Details
- Base model: distilbert-base-uncased
- Task: Text Classification (positive/negative)
- Dataset: Stanford IMDb (5000 train, 1000 test)
- Accuracy: 0.9020 | F1: 0.9019
Usage
from transformers import pipeline
classifier = pipeline("text-classification", model="ashishk-G25AIT2051/MLOps_Final_Group_34")
classifier("This movie was fantastic!")
