Distilbert Goodreads Genres
Fine-tuned DistilBERT model for classifying book reviews into 8 genres.
Source du modèle
- Type
- Other
- Taille du fichier
- 251 MB
- Version
- vRevision 8c2591f
- Mis à jour
- 2 oct. 2026
- Éditeur
- @Pujaniitj
- Téléchargements
- 4
- Source
- Hugging Face
Description de la source
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
Sources
1 source
Sources
1 sourceVérifié 2 oct.
Artefacts du modèle
1 artefact
Artefacts du modèle
1 artefactExtraits de sources
3 extraits
Extraits de sources
3 extraits--- 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
