CoolFace
Modelpublic

Vivek-ML-Projects/distilbert-goodreads-genres

sourceHugging Faceapache-2.0updated 4mo agoView on Hugging Face
0likes7downloads
Model Card

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. 1.Poetry
  2. 2.Children
  3. 3.Comics & Graphic
  4. 4.Fantasy & Paranormal
  5. 5.History & Biography
  6. 6.Mystery, Thriller & Crime
  7. 7.Romance
  8. 8.Young Adult

Usage

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