Distilbert Goodreads Genres

--- 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