Distilbert Goodreads Genres
Fine-tuned DistilBERT model for classifying book reviews into 8 genres.
Fonte do modelo
- Tipo
- Other
- Tamanho do arquivo
- 251 MB
- Versão
- vRevision 8c2591f
- Atualizado
- 2 de out. de 2026
- Publicador
- @Pujaniitj
- Downloads
- 4
- Fonte
- Hugging Face
Descrição da fonte
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
Fontes
1 fonte
Fontes
1 fonteVerificado 2 de out.
Artefatos de modelo
1 artefato
Artefatos de modelo
1 artefatoTrechos de fonte
3 trechos
Trechos de fonte
3 trechos--- 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
