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
9
Safetensors
Model size
0.1B params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ 1 Ask for provider support

Evaluation results