metadata
language: en
license: apache-2.0
tags:
- text-classification
- sentiment-analysis
- bert
- imdb
datasets:
- imdb
metrics:
- accuracy
- f1
BERT IMDB Sentiment Classifier
Fine-tuned bert-base-uncased for binary sentiment classification on the
IMDB movie reviews dataset.
Performance
| Metric | Score |
|---|---|
| Accuracy | 0.934 |
| F1 (pos) | 0.934 |
Usage
```python from transformers import pipeline clf = pipeline("text-classification", model="your-username/bert-imdb-sentiment") clf("This movie was great!") ```
Training
Fine-tuned for 3 epochs, lr=2e-5, batch size 16, on a single GPU.