fiction_predictor_nn

Model Card for aedupuga/fiction_predictor_nn

Model Description

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".

• Model developed by: Anuhya Edupuganti
• Model type: AutoGluon TabularPredictor

Model Sources [optional]

<!-- Provide the basic links for the model. -->

• Dataset: jennifee/HW1-images-dataset

Direct Use

• This model was intended to practice automl implementation on an image dataset

Bias, Risks, and Limitations

• Small data size. cannot to generallised to all existing books on the market.

Training Data:

The model was trained on the augmented split of the "jennifee/HW1-images-dataset".

Evaluation Data:

The model achieved an accuracy of 1.- and a weighted F1 score of 1.0 on the original dataset.

Model Card Contact

Anuhya Edupuganti (Carnegie Mellon Univerity)- [email protected]