Sonny4Sonnix commited on
Commit
940c9eb
·
1 Parent(s): 9bc61f6

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -46,8 +46,7 @@ async def predict(input:model_input):
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])
 
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
  #print(model_input)
52
  df = pd.DataFrame([input])