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

Evaluation results