DistilBERT for Sentiment Analysis - IMDB Movie Reviews
This model is a fine-tuned version of distilbert-base-uncased on the IMDB dataset, trained for sentiment analysis of movie reviews.
Model Details
- Developed by: Shane Reaume (@shane-reaume)
- Model type: Fine-tuned DistilBERT
- Language: English
- License: MIT
- Finetuned from: distilbert-base-uncased
Performance
Evaluation on a balanced test set of 100 examples (50 positive, 50 negative):
| Metric | Value |
|---|---|
| Accuracy | 84.00% |
| F1 Score | 0.8462 |
| Precision | 81.48% |
| Recall | 88.00% |
| Inference Time | ~3.30 ms/example |
Uses
Direct Use
This model can be used directly for sentiment analysis of text, particularly movie reviews.
New Version
- Downloads last month
- 3