You can deploy this model and start earning money today!

Avinash-panda/
bert-sentiment-imdb

Name: bert-sentiment-imdb
Availability: InStock

safetensors

bert

Views

No views yet

BERT for Sentiment Classification (IMDB Dataset)

This model is a fine-tuned version of bert-base-uncased on the IMDB movie reviews dataset for binary sentiment classification (positive / negative).

Model Details

Base model: bert-base-uncased
Fine-tuning task: Sentiment Analysis
Dataset: IMDB (binary labels: positive / negative)
Language(s): English
Framework: PyTorch + Hugging Face Transformers

Training

Batch size: 16
Epochs: 2
Optimizer: AdamW
Learning rate: 5e-5
Max sequence length: 256
Training time: ~X minutes on Google Colab GPU

Evaluation

Test Accuracy: 83.6%

Usage

python

from transformers import pipeline

classifier = pipeline("sentiment-analysis", model="Avinash-panda/bert-sentiment-imdb")

print(classifier("The movie was absolutely fantastic!"))
# [{'label': 'POSITIVE', 'score': 0.98}]