Downloads · 30 days
6
20% of all-time downloads
distilbert-imdb-fine-tuned
Cydonia01/distilbert-imdb-fine-tuned
distilbert-imdb-fine-tuned is a text classification model from Cydonia01. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
This model is a fine-tuned version of distilbert-base-uncased on the IMDB movie reviews dataset for binary sentiment classification (positive/negative). It was trained using Hugging Face's Trainer API.
All-time downloads
30
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
- License
- apache-2.0
- Model type
- distilbert
- Access
- Public
- Created
- Aug 7, 2025
- Updated
- Aug 7, 2025
- SHA
- 44b7c8c2
Try a prompt
Base models
Model card
From the Hugging Face model README
DistilBERT fine-tuned on IMDB
This model is a fine-tuned version of distilbert-base-uncased on the IMDB movie reviews dataset for binary sentiment classification (positive/negative). It was trained using Hugging Face's Trainer API.
Model Details
- Base model: distilbert-base-uncased
- Fine-tuned on: IMDB dataset
- Language: English
- Task: Sentiment classification (positive vs. negative)
- License: Apache-2.0
Usage
You can load and use this model with:
from transformers import pipeline
classifier = pipeline("text-classification", model="Cydonia01/distilbert-imdb-fine-tuned")
classifier("The movie was absolutely amazing!")
Training Details
- Epochs: 3
- Batch size: 16
- Max sequence length: 512
- Optimizer: AdamW
- Loss: CrossEntropyLoss
- Evaluation metric: Accuracy
Evaluation
- Dataset: IMDB test set (25,000 samples)
- Accuracy: 93.46%
Limitations & Biases
This model is trained only on movie reviews and may not generalize well to other domains. It may also inherit biases present in the IMDB dataset.
Author
- Cydonia01
- Task
- Text Classification
- Type
- distilbert
- License
- apache-2.0
- Languages
- en
- Created
- Aug 7, 2025
- Updated
- Aug 7, 2025