fix(Ensembling): stacking now works (although performance is poor)

This commit is contained in:
Mark Aron Szulyovszky
2022-02-19 19:19:02 +01:00
parent 7c08a87243
commit 229f16c4b3
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -98,7 +98,7 @@ def get_model(model_name: str) -> Model:
"validation_type": "split",
"train_ratio": 0.75,
"shuffle": False,
"stratify": True
"stratify": True,
},
eval_metric="f1",
)