sentiment-bert / README.md
ManavDhayeCoder's picture
READ
8cbd529 verified
metadata
language:
  - en
tags:
  - sentiment-analysis
  - text-classification
  - bert
  - manav
  - ManavDhayeCoder/sentiment-bert
  - ManavDhaye
pipeline_tag: text-classification
base_model:
  - google-bert/bert-base-uncased
datasets:
  - imdb
library_name: transformers
widget:
  - text: This movie was amazing!
  - text: Worst movie I have ever seen.
model-index:
  - name: sentiment-bert
    results: []
metrics:
  - accuracy

πŸ“˜ 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 β†’ negative
  • LABEL_1 β†’ positive

πŸ”₯ Quick Usage Example

from transformers import pipeline

clf = pipeline("text-classification", model="ManavDhayeCoder/sentiment-bert")

print(clf("This movie was amazing!"))