File size: 1,130 Bytes
870e323 | 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 | ---
language: en
license: apache-2.0
base_model: bert-base-uncased
tags:
- sentiment-analysis
- imdb
- text-classification
widget:
- text: "This movie was absolutely fantastic! Best film I've seen all year."
---
# BERT Fine-tuned on IMDB for Sentiment Analysis
Fine-tuned from `bert-base-uncased` on the Stanford IMDB dataset for binary sentiment classification.
## Training Details
| Parameter | Value |
|-----------|-------|
| Base model | bert-base-uncased |
| Learning rate | 2e-5 |
| Batch size | 4 |
| Epochs | 2 |
| Max sequence length | 512 |
## Usage
```python
from transformers import BertForSequenceClassification, BertTokenizer
tokenizer = BertTokenizer.from_pretrained("COMP6713bert/imdb-bert-sentiment")
model = BertForSequenceClassification.from_pretrained("COMP6713bert/imdb-bert-sentiment")
inputs = tokenizer("This movie was great!", return_tensors="pt", truncation=True, max_length=512)
with torch.no_grad():
outputs = model(**inputs)
predicted = torch.argmax(outputs.logits, dim=-1).item()
print("Positive" if predicted == 1 else "Negative")
```
## Labels
- 0: Negative
- 1: Positive |