| 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. | |