mirror of
https://github.com/webclinic017/drift.git
synced 2026-07-27 18:57:55 +00:00
fix(Ensembling): stacking now works (although performance is poor)
This commit is contained in:
@@ -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
@@ -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",
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user