Commit
·
940c9eb
1
Parent(s):
9bc61f6
Update main.py
Browse files
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])
|