Downloads · 30 days
24
3% of all-time downloads
distilbert-imdb-mlflow
juliensimon/distilbert-imdb-mlflow
distilbert-imdb-mlflow is a text classification model from juliensimon. 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.
A DistilBERT model fine-tuned on the IMDB dataset for sentiment analysis, with MLflow experiment tracking.
All-time downloads
788
Public
Parameters
65.8M
526 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin263 MB · 50%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- distilbert
- Access
- Public
- Created
- Jul 15, 2022
- Updated
- Mar 22, 2026
- SHA
- cfd24cb4
Try a prompt
Base models
Datasets
Model card
From the Hugging Face model README
DistilBERT IMDB — MLflow Integration
A DistilBERT model fine-tuned on the IMDB dataset for sentiment analysis, with MLflow experiment tracking.
Video walkthrough: Log with MLflow and Hugging Face Transformers
Model Details
| Detail | Value |
|---|---|
| Base model | distilbert-base-cased |
| Task | Binary sentiment classification |
| Dataset | IMDB (50K movie reviews) |
| License | Apache 2.0 |
Training Configuration
| Parameter | Value |
|---|---|
| Learning rate | 5e-05 |
| Batch size | 8 (train & eval) |
| Optimizer | Adam (β₁=0.9, β₂=0.999, ε=1e-08) |
| LR scheduler | Linear |
| Epochs | 1 |
Usage
from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="juliensimon/distilbert-imdb-mlflow")
classifier("This movie was absolutely fantastic!")
Viewing MLflow Logs
MLflow training logs are included in the repository. To view:
pip install mlflow
mlflow ui
Framework Versions
- Transformers 4.20.1
- PyTorch 1.10.0
- Datasets 2.3.2
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- License
- apache-2.0
- Created
- Jul 15, 2022
- Updated
- Mar 22, 2026