File size: 982 Bytes
5fe223b cb6896e 5fe223b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | # 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)- aedupuga@andrew.cmu.edu |