Downloads · 30 days
9
33% of all-time downloads
imdb-sentiment-distilbert
hadsaw/imdb-sentiment-distilbert
imdb-sentiment-distilbert is a text classification model from hadsaw. Use it when you need a label for a piece of text.
This model is a fine-tuned version of distilbert-base-uncased on the IMDB dataset for sentiment analysis.
All-time downloads
27
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
- Model type
- distilbert
- Access
- Public
- Created
- Feb 3, 2025
- Updated
- Feb 3, 2025
- SHA
- 55df0ad4
Try a prompt
Datasets
Model card
From the Hugging Face model README
IMDB Sentiment Analysis Model
This model is a fine-tuned version of distilbert-base-uncased on the IMDB dataset for sentiment analysis.
Model description
The model was trained on the IMDB movie reviews dataset to classify movie reviews as either positive (1) or negative (0).
Training procedure
- Model: DistilBERT (distilbert-base-uncased)
- Training Data: IMDB Dataset (40,000 training samples)
- Validation Data: 5,000 samples
- Test Data: 5,000 samples
- Epochs: 2
- Batch Size: 32
- Learning Rate: 2e-5
- Max Sequence Length: 256 tokens
Usage
- Task
- Text Classification
- Type
- distilbert
- Languages
- en
- Created
- Feb 3, 2025
- Updated
- Feb 3, 2025