Sonny4Sonnix commited on
Commit
9bc61f6
·
1 Parent(s): 930e760

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
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])