JD45-d/distilbert-goodreads-genre-classifier

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 transformers import pipeline

classifier = pipeline(
    "text-classification", 
    model="JD45-d/distilbert-goodreads-genre-classifier"
)

result = classifier("This was the most romantic book I have ever read!")
print(result)