Sentiment-BERT-IMDB
A BERT-based model fine-tuned on the IMDB movie reviews dataset for binary sentiment classification (positive/negative). This model is intended for quick deployment and practical use in applications like review analysis, recommendation systems, and content moderation.
Model Details
- Architecture:
bert-base-uncased
- Task: Sentiment classification (positive vs. negative)
- Dataset: IMDB
- Classes:
positive
,negative
- Tokenizer:
bert-base-uncased
How to Use
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
model = AutoModelForSequenceClassification.from_pretrained("HrishikeshDeore/sentiment-bert-imdb")
tokenizer = AutoTokenizer.from_pretrained("HrishikeshDeore/sentiment-bert-imdb")
nlp = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
result = nlp("This movie was absolutely fantastic!")
print(result)
- Downloads last month
- 11