DistilBERT IMDB Sentiment Classifier
This model is a fine-tuned version of distilbert-base-uncased on the IMDB dataset for sentiment analysis.
Task
Binary classification of movie reviews:
Training Details
- Base model: DistilBERT
- Dataset: IMDB
- Training samples: 2000
- Epochs: 1
Usage
from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="nidhic0301/distilbert-imdb-sentiment-classifier")
classifier("This movie is amazing!")
