You can deploy this model and start earning money today!

Khushon123/
imdb-sentiment-model

Name: imdb-sentiment-model
Availability: InStock

safetensors

distilbert

Views

No views yet

IMDB Sentiment Classifier (DistilBERT)

This is a fine-tuned DistilBERT model trained on a subset of the IMDB movie review dataset for binary sentiment classification.

Model Details

Base Model: distilbert-base-uncased
Fine-tuned On: IMDB reviews (custom subset: 2,000 train / 500 test)
Labels: POSITIVE, NEGATIVE
Framework: PyTorch, Transformers
Training Time: ~3 minutes (NVIDIA RTX 4080 Laptop GPU)

Dataset

Source: https://huggingface.co/datasets/imdb
Preprocessed: Truncated to 2000 training + 500 testing samples for speed

Metrics

Epoch	Accuracy
1	84%
2	88%

Evaluation metric: Accuracy
Optimizer: AdamW with weight decay
Scheduler: linear

How to Use

python

from transformers import pipeline

classifier = pipeline("sentiment-analysis", model="khushon123/imdb-sentiment-model")
classifier("This movie was absolutely fantastic!")

Demo

Try the model in a live web app:

➡️ https://huggingface.co/spaces/khushon123/imdb-sentiment-app

Files

config.json
pytorch_model.bin
tokenizer_config.json
vocab.txt

License

MIT License

Author

Created by https://huggingface.co/khushon123

Khushon123/imdb-sentiment-model