bert-sentiment-imdb
This is a fine-tuned BERT model for sentiment analysis on the IMDB dataset.
- Base Model:
bert-base-uncased - Dataset: IMDB
- Task: Sentiment Analysis (Binary)
- Usage:
from transformers import pipeline
classifier = pipeline("sentiment-analysis"model="hmuegyi/review-classifier")
# Test
result = classifier("This movie is nice!")
print(result)
- Downloads last month
- 7
Model tree for hmuegyi/review-classifier
Base model
google-bert/bert-base-uncased