gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-14 19:38:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(Sweep): try to filter out some not great models (#144)

Browse Source 
* feat(Sweep): try to filter out some not great models

* fix(Sweep): yaml

* fix(Sweep): yaml

* fix(Config): remove some models that do not perform well
This commit is contained in:
Mark Aron Szulyovszky
2022-01-11 10:33:18 +01:00
committed by GitHub
parent 2c03959315
commit 54ea59c0cf
3 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
config/config.py
+1 -1
View File
@@ -73,7 +73,7 @@ def get_default_ensemble_config() -> tuple[dict, dict, dict]:
)
regression_models = ["Lasso", "KNN", "RF"]
classification_models = ['LR_two_class', 'SVC', 'KNN', 'CART', 'NB', 'AB', 'RF', 'XGB_two_class', 'LGBM', 'StaticMom']
classification_models = ["LR_two_class", "LDA", "NB", "RF", "XGB_two_class", "LGBM", "StaticMom"]
meta_labeling_models = ['LR_two_class', 'LGBM']
ensemble_model = 'Average'