File size: 1,417 Bytes
000a0ab |
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 36 37 38 39 40 41 42 |
---
# For reference on model card metadata, see the spec: https://github.com/huggingface/hub-docs/blob/main/modelcard.md?plain=1
# Doc / guide: https://huggingface.co/docs/hub/model-cards
{}
---
# Model Card for aedupuga/recommendation_predictor
### Model Description
This is an AutoGluon Tabular AutoML implementation on a tabular dataset recording different features of the book. The model predicts whether the author would 'Recommend' or 'Not Recommend' a book based on given features.
- **Model developed by:** Anuhya Edupuganti
- **Model type:** AutoGluon TabularPredictor
### Model Sources [optional]
<!-- Provide the basic links for the model. -->
- **Dataset:** jennifee/HW1-tabular-dataset
### Direct Use
- This model was intended to practice automl implementation on a tabular dataset
## Bias, Risks, and Limitations
- Small data size.
- Personal preference of the dataset creator in classification.
## Training Data:
The model was trained on the augmented split of the "jennifee/HW1-tabular-dataset" The data includes features such as FictionorNonfiction, NumPages, ThicknessInches, and ReadUnfinishedorUnread, with the target variable being RecommendtoEveryone (yes or no).
## Evaluation Data:
The model achieved an accuracy of 0.5000 and a weighted F1 score of 0.5212 on the original dataset.
## Model Card Contact
Anuhya Edupuganti (Carnegie Mellon Univerity)- aedupuga@andrew.cmu.edu |