Downloads · 30 days
0
image-autogluon-predictor
aedupuga/image-autogluon-predictor
image-autogluon-predictor is a machine learning model from aedupuga. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
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".
Access
Public
Updated Sep 23, 2025
Repo size
86.5 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pkl44.9 MB · 52%
At a glance
- Access
- Public
- Created
- Sep 23, 2025
- Updated
- Sep 23, 2025
- SHA
- 3e4e5fe9
Model card
From the Hugging Face model README
Model Card for aedupuga/image-autogluon-predictor
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.000 and a weighted F1 score of 1.000 on the original dataset.
Model Card Contact
Anuhya Edupuganti (Carnegie Mellon Univerity)- aedupuga@andrew.cmu.edu
- Created
- Sep 23, 2025
- Updated
- Sep 23, 2025