judge-books-cover

judge-books-cover

👨‍💻 Machine Learning Task 👨‍💻

Deliverable

1. Documented model notebook
   • Source .pdf
   • Source .ipynb

<script src="https://gist.github.com/PseudoCodeNerd/06f623772724521c6151774907aed5e8.js"></script>

---

Judge books by their cover using Flux.jl

Task Description

Create a machine learning model to predict the category of a book from its cover image

This task is inspired by this paper. Your task is to use the Flux machine learning library to predict the category of books in this dataset based on their cover images.

You can find the Flux documentation here and sample models for image categorization in the model zoo. We recommend starting with a simple model like this one and then optionally using a more complex one if you are interested.

P.S: I made a complex model and was able to near the accuracy put forward in the original paper . Dope