Downloads · 30 days
5
6% of all-time downloads
vj-distilbert-imdb-sentiment
vijayjadhavaug/vj-distilbert-imdb-sentiment
vj-distilbert-imdb-sentiment is a text classification model from vijayjadhavaug. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
This is a fine-tuned DistilBERT model for sentiment analysis on the IMDB dataset. It classifies reviews into Positive or Negative sentiments.
All-time downloads
78
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- distilbert
- Access
- Public
- Created
- Feb 15, 2025
- Updated
- Feb 15, 2025
- SHA
- d200e97a
Try a prompt
Model card
From the Hugging Face model README
🏆 DistilBERT IMDB Sentiment Classifier
This is a fine-tuned DistilBERT model for sentiment analysis on the IMDB dataset.
It classifies reviews into Positive or Negative sentiments.
📖 Model Details
- Base Model:
distilbert-base-uncased - Dataset: IMDB Reviews
- Fine-Tuning Epochs: 3
- Batch Size: 16
- Evaluation Loss: 0.291
🚀 Usage
from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="vj/distilbert-imdb-sentiment")
result = classifier("The movie was absolutely fantastic!")
print(result) # [{'label': 'POSITIVE', 'score': 0.98}]
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- License
- apache-2.0
- Created
- Feb 15, 2025
- Updated
- Feb 15, 2025