Update app.py
Browse files
app.py
CHANGED
@@ -13,47 +13,98 @@ loaded_model = pickle.load(open("db_xgb.pkl", 'rb'))
|
|
13 |
explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
|
14 |
|
15 |
# Define mapping functions
|
16 |
-
def
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
|
|
21 |
|
22 |
def map_HighChol(value):
|
23 |
-
|
|
|
|
|
|
|
|
|
24 |
|
25 |
-
def
|
26 |
-
|
|
|
|
|
|
|
|
|
27 |
|
28 |
def map_Smoker(value):
|
29 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
30 |
|
31 |
def map_Stroke(value):
|
32 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
|
34 |
def map_HeartDiseaseorAttack(value):
|
35 |
-
|
|
|
|
|
|
|
|
|
36 |
|
37 |
def map_PhysActivity(value):
|
38 |
-
|
39 |
-
|
|
|
|
|
|
|
|
|
40 |
def map_Fruits(value):
|
41 |
-
|
42 |
-
|
|
|
|
|
|
|
|
|
43 |
def map_Veggies(value):
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
|
|
48 |
|
49 |
def map_AnyHealthcare(value):
|
50 |
-
|
|
|
|
|
|
|
|
|
51 |
|
52 |
def map_NoDocbcCost(value):
|
53 |
-
|
|
|
|
|
|
|
|
|
|
|
54 |
|
55 |
-
def map_DiffWalk(value):
|
56 |
-
return map_bool(value)
|
57 |
|
58 |
# Create the main function for server
|
59 |
def main_func(HighBP, HighChol, CholCheck, BMI, Smoker, Stroke, HeartDiseaseorAttack, PhysActivity, Fruits, Veggies, HvyAlcoholConsump, AnyHealthcare, NoDocbcCost, GenHlth, MentHlth, PhysHlth, DiffWalk, Sex, Age, Education, Income):
|
|
|
13 |
explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
|
14 |
|
15 |
# Define mapping functions
|
16 |
+
def map_CholCheck(value):
|
17 |
+
mapping = {'No': 0, 'Yes': 1}
|
18 |
+
return mapping[value]
|
19 |
+
Chol_Checked = 'Yes'
|
20 |
+
numeric_value = map_CholCheck(Chol_Checked)
|
21 |
+
print(numeric_value)
|
22 |
|
23 |
def map_HighChol(value):
|
24 |
+
mapping = {'No': 0, 'Yes': 1}
|
25 |
+
return mapping[value]
|
26 |
+
HighChol_Checked = 'Yes'
|
27 |
+
numeric_value = map_HighChol(HighChol_Checked)
|
28 |
+
print(numeric_value)
|
29 |
|
30 |
+
def map_DiffWalk(value):
|
31 |
+
mapping = {'No': 0, 'Yes': 1}
|
32 |
+
return mapping[value]
|
33 |
+
DiffWalk_Checked = 'Yes'
|
34 |
+
numeric_value = map_DiffWalk(DiffWalk_Checked)
|
35 |
+
print(numeric_value)
|
36 |
|
37 |
def map_Smoker(value):
|
38 |
+
mapping = {'No': 0, 'Yes': 1}
|
39 |
+
return mapping[value]
|
40 |
+
Smoker_Checked = 'Yes'
|
41 |
+
numeric_value = map_Smoker(Smoker_Checked)
|
42 |
+
print(numeric_value)
|
43 |
+
|
44 |
+
def map_HvyAlcoholConsump(value):
|
45 |
+
mapping = {'No': 0, 'Yes': 1}
|
46 |
+
return mapping[value]
|
47 |
+
HvyAlcoholConsump_Checked = 'Yes'
|
48 |
+
numeric_value = map_HvyAlcoholConsump(HvyAlcoholConsump_Checked)
|
49 |
+
print(numeric_value)
|
50 |
|
51 |
def map_Stroke(value):
|
52 |
+
mapping = {'No': 0, 'Yes': 1}
|
53 |
+
return mapping[value]
|
54 |
+
Stroke_Checked = 'Yes'
|
55 |
+
numeric_value = map_Stroke(Stroke_Checked)
|
56 |
+
print(numeric_value)
|
57 |
+
|
58 |
+
def map_HighBP(value):
|
59 |
+
mapping = {'No': 0, 'Yes': 1}
|
60 |
+
return mapping[value]
|
61 |
+
HighBP_Checked = 'Yes'
|
62 |
+
numeric_value = map_HighBP(HighBP_Checked)
|
63 |
+
print(numeric_value)
|
64 |
|
65 |
def map_HeartDiseaseorAttack(value):
|
66 |
+
mapping = {'No': 0, 'Yes': 1}
|
67 |
+
return mapping[value]
|
68 |
+
HeartDiseaseorAttack_Checked = 'Yes'
|
69 |
+
numeric_value = map_HeartDiseaseorAttack(HeartDiseaseorAttack_Checked)
|
70 |
+
print(numeric_value)
|
71 |
|
72 |
def map_PhysActivity(value):
|
73 |
+
mapping = {'No': 0, 'Yes': 1}
|
74 |
+
return mapping[value]
|
75 |
+
PhysActivity_Checked = 'Yes'
|
76 |
+
numeric_value = map_PhysActivity(PhysActivity_Checked)
|
77 |
+
print(numeric_value)
|
78 |
+
|
79 |
def map_Fruits(value):
|
80 |
+
mapping = {'No': 0, 'Yes': 1}
|
81 |
+
return mapping[value]
|
82 |
+
Fruits_Checked = 'Yes'
|
83 |
+
numeric_value = map_Fruits(Fruits_Checked)
|
84 |
+
print(numeric_value)
|
85 |
+
|
86 |
def map_Veggies(value):
|
87 |
+
mapping = {'No': 0, 'Yes': 1}
|
88 |
+
return mapping[value]
|
89 |
+
Veggies_Checked = 'Yes'
|
90 |
+
numeric_value = map_Veggies(Veggies_Checked)
|
91 |
+
print(numeric_value)
|
92 |
|
93 |
def map_AnyHealthcare(value):
|
94 |
+
mapping = {'No': 0, 'Yes': 1}
|
95 |
+
return mapping[value]
|
96 |
+
AnyHealthcare_Checked = 'Yes'
|
97 |
+
numeric_value = map_AnyHealthcare(AnyHealthcare_Checked)
|
98 |
+
print(numeric_value)
|
99 |
|
100 |
def map_NoDocbcCost(value):
|
101 |
+
mapping = {'No': 0, 'Yes': 1}
|
102 |
+
return mapping[value]
|
103 |
+
map_NoDocbcCost_Checked = 'Yes'
|
104 |
+
numeric_value = map_NoDocbcCost(map_NoDocbcCost_Checked)
|
105 |
+
print(numeric_value)
|
106 |
+
|
107 |
|
|
|
|
|
108 |
|
109 |
# Create the main function for server
|
110 |
def main_func(HighBP, HighChol, CholCheck, BMI, Smoker, Stroke, HeartDiseaseorAttack, PhysActivity, Fruits, Veggies, HvyAlcoholConsump, AnyHealthcare, NoDocbcCost, GenHlth, MentHlth, PhysHlth, DiffWalk, Sex, Age, Education, Income):
|