Commit
·
9bc61f6
1
Parent(s):
930e760
Update main.py
Browse files
main.py
CHANGED
@@ -43,11 +43,11 @@ async def predict(input:model_input):
|
|
43 |
'Blood_Work_Result_2',' Blood_Work_Result_3', 'Body_mass_index',
|
44 |
'Blood_Work_Result_4', 'Age']]
|
45 |
|
46 |
-
XGB= Pipeline([
|
47 |
("col_trans", full_pipeline),
|
48 |
("feature_selection", SelectKBest(score_func=f_classif, k='all')),
|
49 |
("model", BaggingClassifier(base_estimator=XGBClassifier(random_state=42)))
|
50 |
-
])
|
51 |
|
52 |
#print(model_input)
|
53 |
df = pd.DataFrame([input])
|
|
|
43 |
'Blood_Work_Result_2',' Blood_Work_Result_3', 'Body_mass_index',
|
44 |
'Blood_Work_Result_4', 'Age']]
|
45 |
|
46 |
+
XGB= Pipeline([
|
47 |
("col_trans", full_pipeline),
|
48 |
("feature_selection", SelectKBest(score_func=f_classif, k='all')),
|
49 |
("model", BaggingClassifier(base_estimator=XGBClassifier(random_state=42)))
|
50 |
+
])
|
51 |
|
52 |
#print(model_input)
|
53 |
df = pd.DataFrame([input])
|