Beta
Explore
Marketplace
Neural Labs
Playground
Wallet
Docs
distilbert-goodreads-genres – AI Model by Pujaniitj | AlphaNeural AI
You can deploy this model and start earning money today!
Pujaniitj
/
distilbert-goodreads-genres
like
0
safetensors
distilbert
text-classification
goodreads
genre-classification
apache-2.0
us
Views
No views yet
Model card
Files and Versions
Community
API
Deploy
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
