File size: 701 Bytes
a64948a | 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 | ---
license: mit
---
# IMDb Sentiment Analysis - Fine-tuned BERT Model
This project fine-tunes a pre-trained BERT model from Hugging Face on the IMDb movie reviews dataset for binary sentiment classification (positive/negative).
## Project Overview
- Task: Sentiment Analysis (Binary Classification)
- Dataset: IMDb
- Model: bert-base-uncased fine-tuned using Transformers and PyTorch
- Accuracy: About 92.8% on validation data
## Sample Usage
### Load the model and tokenizer
```python
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("saved_model")
model = AutoModelForSequenceClassification.from_pretrained("saved_model")
|