SuganyaP/quick-distilbert-imdb

Quick DistilBERT IMDB Sentiment Classifier

This is a fine-tuned DistilBERT model for sentiment analysis on the IMDB movie reviews dataset.
The model classifies reviews as positive or negative.

Model Details

• Base model: distilbert-base-uncased
• Dataset: IMDB (cleaned train/test splits)
• Task: Sentiment classification (binary)
• Framework: Hugging Face Transformers

Training

• Optimized DistilBERT on IMDB dataset
• Used standard text classification head
• Training args saved in training_args.bin

Evaluation

Accuracy and F1-score on the IMDB test set:
(Add numbers from your eval_report.txt here)

Misclassified examples are available in misclassified_examples.csv.

How to Use

from transformers import pipeline

model_id = "SuganyaP/quick-distilbert-imdb"
classifier = pipeline("sentiment-analysis", model=model_id)

print(classifier("This movie was excellent!"))