aedupuga commited on
Commit
000a0ab
·
verified ·
1 Parent(s): 37d64a1

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +44 -0
README.md ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ # For reference on model card metadata, see the spec: https://github.com/huggingface/hub-docs/blob/main/modelcard.md?plain=1
3
+ # Doc / guide: https://huggingface.co/docs/hub/model-cards
4
+ {}
5
+ ---
6
+
7
+ # Model Card for aedupuga/recommendation_predictor
8
+
9
+ ### Model Description
10
+
11
+
12
+ 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.
13
+
14
+
15
+ - **Model developed by:** Anuhya Edupuganti
16
+ - **Model type:** AutoGluon TabularPredictor
17
+
18
+
19
+ ### Model Sources [optional]
20
+
21
+ <!-- Provide the basic links for the model. -->
22
+ - **Dataset:** jennifee/HW1-tabular-dataset
23
+
24
+
25
+ ## Uses
26
+
27
+ ### Direct Use
28
+ - This model was intended to practice automl implementation on a tabular dataset
29
+
30
+ ## Bias, Risks, and Limitations
31
+ - Small data size.
32
+ - Personal preference of the dataset creator in classification.
33
+
34
+ ## Training Data:
35
+
36
+ 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).
37
+
38
+ ## Evaluation Data:
39
+
40
+ The model achieved an accuracy of 0.5000 and a weighted F1 score of 0.5212 on the original dataset.
41
+
42
+ ## Model Card Contact
43
+
44
+ Anuhya Edupuganti (Carnegie Mellon Univerity)- aedupuga@andrew.cmu.edu