Modelpublic

kamalchaurasia-iitj/goodreads-reviews-genres

sourceHugging Facemitupdated 4mo agoView on Hugging Face

0likes17downloads

Model Card

Goodreads Book-Genre Classifier

Fine-tuned `roberta-base` for 8-class book-genre classification on the UCSD Goodreads Reviews dataset.

Genres

—children
—comics_graphic
—fantasy_paranormal
—history_biography
—mystery_thriller_crime
—poetry
—romance
—young_adult

Training details

Parameter	Value
Base model	`roberta-base`
Epochs	`3`
Learning rate	`5e-05`
Train batch size	`10`
Warmup ratio	`None`
Weight decay	`0.01`

Evaluation (test set)

Metric	Score
Accuracy	0.6162
Macro F1	0.6092
Loss	2.2351

Usage

python

from transformers import pipeline

classifier = pipeline(
    "text-classification",
    model="kamalchaurasia-iitj/goodreads-reviews-genres",
    tokenizer="kamalchaurasia-iitj/goodreads-reviews-genres",
)

result = classifier("I loved the mystery and suspense in this thrilling novel.")
print(result)  # [{'label': 'mystery_thriller_crime', 'score': 0.98}]

Citation

@misc{goodreads-genre-classifier,
  author = {Kamalkumar Chaurasia},
  title  = {Goodreads Book-Genre Classifier (roberta-base)},
  year   = {2026},
  url    = {https://huggingface.co/kamalchaurasia-iitj/goodreads-reviews-genres}
}