Downloads · 30 days
6
1% of all-time downloads
distilbert-goodreads-genres
Pujaniitj/distilbert-goodreads-genres
distilbert-goodreads-genres is a text classification model from Pujaniitj. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
Fine-tuned DistilBERT model for classifying book reviews into 8 genres.
All-time downloads
404
Public
Parameters
65.8M
5.8 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors263 MB · 100%
At a glance
- Task
- Text Classification
- License
- apache-2.0
- Model type
- distilbert
- Access
- Public
- Created
- May 10, 2026
- Updated
- May 26, 2026
- SHA
- 8c2591fb
Try a prompt
Model card
From the Hugging Face model README
DistilBERT Goodreads Genre Classification
Model Description
Fine-tuned DistilBERT model for classifying book reviews into 8 genres.
Training Hyperparameters
- Epochs: 3
- Batch Size: 16
- Learning Rate: 5e-05
- Optimizer: AdamW
Evaluation Results
| Metric | Score |
|---|---|
| Accuracy | 0.5956 |
| F1 Score | 0.5988 |
| Eval Loss | 1.1631 |
Links
- W&B Run: https://wandb.ai/pujaniitj-iit-jodpur/mlops-assignment2/runs/l3ka09fm
- Kaggle Notebook: https://www.kaggle.com/code/g25ait2076/mlops-final
- GitHub Repository: https://github.com/pujaniitj/Complete-mlops-workflow
- Task
- Text Classification
- Type
- distilbert
- License
- apache-2.0
- Created
- May 10, 2026
- Updated
- May 26, 2026