mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-15 11:58:07 +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(
|
StackingClassifier(
|
||||||
[(m.name, m) for m in estimators],
|
[(m.name, m) for m in estimators],
|
||||||
final_estimator=estimators[0],
|
final_estimator=estimators[0],
|
||||||
cv=TimeSeriesSplit(gap=100),
|
cv=5,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
|
|||||||
+1
-1
@@ -98,7 +98,7 @@ def get_model(model_name: str) -> Model:
|
|||||||
"validation_type": "split",
|
"validation_type": "split",
|
||||||
"train_ratio": 0.75,
|
"train_ratio": 0.75,
|
||||||
"shuffle": False,
|
"shuffle": False,
|
||||||
"stratify": True
|
"stratify": True,
|
||||||
},
|
},
|
||||||
eval_metric="f1",
|
eval_metric="f1",
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user