nidhic0301/distilbert-imdb-sentiment-classifier

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:

• Positive
• Negative

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!")

Note - Do checkout a more accurate and fine tuned sentiment analyser for IMDB reviews here - https://huggingface.co/nidhic0301/distilbert-imdb-sentiment-classifier-lora