File size: 957 Bytes
d5334dc a24adf8 d5334dc a24adf8 d5334dc a24adf8 d5334dc a24adf8 | 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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | ---
language: en
license: apache-2.0
tags:
- text-classification
- sentiment-analysis
- bert
- imdb
datasets:
- imdb
metrics:
- accuracy
- f1
---
# BERT Fine-tuned on IMDB Sentiment Analysis
## Model Description
This model is a fine-tuned version of `bert-base-uncased`
on the IMDB movie reviews dataset for sentiment analysis.
## Training Details
- Base Model: bert-base-uncased
- Dataset: IMDB (2000 train, 500 test samples)
- Epochs: 3
- Learning Rate: 2e-5
- Batch Size: 16
- Framework: HuggingFace Transformers
## Results
| Metric | Score |
|----------|-------|
| Accuracy | ~88% |
| F1 Score | ~0.88 |
## Usage
```python
from transformers import pipeline
classifier = pipeline(
'sentiment-analysis',
model='your-hf-username/bert-imdb-sentiment'
)
result = classifier("This movie was absolutely amazing!")
print(result)
# [{'label': 'POSITIVE', 'score': 0.98}]
```
## Labels
- LABEL_0 → Negative 😠
- LABEL_1 → Positive 😊
|