metadata
license: mit
language:
- en
base_model:
- google-bert/bert-base-uncased
pipeline_tag: text-classification
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)