File size: 640 Bytes
7ff2b72 b6ac5a7 7ff2b72 b6ac5a7 7ff2b72 b6ac5a7 7ff2b72 b6ac5a7 7ff2b72 b6ac5a7 7ff2b72 b6ac5a7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | ---
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.
|