This is an AutoGluon Image AutoML NN implementation on a image dataset containing cover images of books. The model predicts whether the book is "fiction" or "non-fiction".
The model was trained on the augmented split of the "jennifee/HW1-images-dataset".
The model achieved an accuracy of 1.000 and a weighted F1 score of 1.000 on the original dataset.
Anuhya Edupuganti (Carnegie Mellon Univerity)-
aedupuga@andrew.cmu.edu
