Distilbert Goodreads Genres
Fine-tuned DistilBERT model for classifying book reviews into 8 genres.
Fuente del modelo
- Tipo
- Other
- Tamaño del archivo
- 251 MB
- Versión
- vRevision 8c2591f
- Actualizado
- 2 oct 2026
- Publicador
- @Pujaniitj
- Descargas
- 4
- Fuente
- Hugging Face
Descripción de la fuente
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
Fuentes
1 fuente
Fuentes
1 fuenteVerificado 2 oct
Artefactos del modelo
1 artefacto
Artefactos del modelo
1 artefactoExtractos de fuentes
3 extractos
Extractos de fuentes
3 extractos--- license: apache-2.0 tags: - text-classification - distilbert - goodreads - genre-classification metrics: - accuracy - f1 --- # 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
Source context: 4 downloads · 0 likes · Pipeline text-classification · Repo Pujaniitj/distilbert-goodreads-genres
Source context: 1 downloads · 0 likes · Pipeline text-classification · Repo Pujaniitj/distilbert-goodreads-genres
