gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-18 13:28:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(Ensembling): added possiblity of stacking models

Browse Source
This commit is contained in:
Mark Aron Szulyovszky
2022-02-19 18:55:17 +01:00
parent cfb65c135e
commit 7c08a87243
4 changed files with 67 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
config/types.py
+1 -1
View File
@@ -29,7 +29,7 @@ class RawConfig(BaseModel):
labeling: Literal["two_class", "three_class_balanced", "three_class_imbalanced"]
forecasting_horizon: int
save_models: bool
ensembling_method: Literal["voting_soft", "voting_hard"]
ensembling_method: Literal["voting_soft", "stacking"]
directional_models: list[str]
meta_models: list[str]