Vivek-ML-Projects/distilbert-goodreads-genres

DistilBERT Goodreads Genre Classifier

This model classifies book reviews into 8 genres using the UCSD Goodreads dataset.

Model Description

• Base Model: distilbert-base-cased
• Task: Multi-class text classification
• Number of Labels: 8

Supported Genres:

1. Poetry
2. Children
3. Comics & Graphic
4. Fantasy & Paranormal
5. History & Biography
6. Mystery, Thriller & Crime
7. Romance
8. Young Adult

Usage

from huggingface_hub import login
 
# HF_TOKEN was already loaded from Kaggle Secrets (see Task 4)
login(token=HF_TOKEN)
 
# Push model and tokenizer
model.push_to_hub("Vivek-ML-Projects/distilbert-goodreads-genres")
tokenizer.push_to_hub("Vivek-ML-Projects/distilbert-goodreads-genres")
 
# Record the link in W&B
wandb.run.summary["huggingface_model"] = \
    "https://huggingface.co/Vivek-ML-Projects/distilbert-goodreads-genres"