Skip to content

This project focuses on developing a multi-label genre classifier using DistilBERT. The primary goal is to scrape book descriptions and genre labels from Smashwords.com and use this data to train a DistilBERT model that can accurately predict multiple genres for a given book description.

License

Notifications You must be signed in to change notification settings

krishnaraketla/multi-label-genre-classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

multi-label-genre-classifier

This project focuses on developing a multi-label genre classifier using DistilBERT. The primary goal is to scrape book descriptions and genre labels from Smashwords.com and use this data to train a DistilBERT model that can accurately predict multiple genres for a given book description.

About

This project focuses on developing a multi-label genre classifier using DistilBERT. The primary goal is to scrape book descriptions and genre labels from Smashwords.com and use this data to train a DistilBERT model that can accurately predict multiple genres for a given book description.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published