π BERT Sentiment Analysis Model (Fine-Tuned on IMDB)
This model is a fine-tuned version of google-bert/bert-base-uncased, trained on the IMDB movie reviews dataset for binary sentiment classification.
It predicts whether text expresses negative or positive sentiment.
This model is hosted by @ManavDhayeCoder.
π Model Overview
| Property | Value |
|---|---|
| Base model | google-bert/bert-base-uncased |
| Task | Sentiment Analysis (Sequence Classification) |
| Labels | negative / positive |
| Dataset | IMDB |
| Library | Hugging Face Transformers |
| Format | model.safetensors |
The model has two classes:
LABEL_0β negativeLABEL_1β positive
π₯ Quick Usage Example
from transformers import pipeline
clf = pipeline("text-classification", model="ManavDhayeCoder/sentiment-bert")
print(clf("This movie was amazing!"))
- Downloads last month
- 33
Model tree for ManavDhayeCoder/sentiment-bert
Base model
google-bert/bert-base-uncased