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
- Downloads last month
- 381