Atreyee-Halder/distilbert-goodreads-genres

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: 3e-5
• Optimizer: AdamW

Evaluation Results

Metric	Score
Accuracy	0.5919
F1 Score	0.5935
Eval Loss	2.2633

Links

• W&B: https://wandb.ai/g25ait2023-iit-jodhpur/mlops-assignment2
• Kaggle: https://www.kaggle.com/code/ahalderg25ait2023/mlops-assignment-kaggle
• GitHub: https://github.com/halderatreyee-hash/MLOPS-ASSIGNMENT2_Kaggle