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
@@ -59,7 +59,7 @@ def __preprocess_model_config(model_config: dict) -> dict:
StackingClassifier(
[(m.name, m) for m in estimators],
final_estimator=estimators[0],
cv=TimeSeriesSplit(gap=100),
cv=5,
)
)
else:
+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",
)