gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-07-27 18:57:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(Models): added lightGBM, moved other models to separate files (#128)

Browse Source 
* feat(Models): added lightGBM, moved other models to separate files

* feat(Models): added non-working statsmodel wrapper

* fix(Models): added work-in-progress comment to StatsModels
This commit is contained in:
Mark Aron Szulyovszky
2022-01-08 12:02:42 +01:00
committed by GitHub
parent 6982187872
commit fc5eba4e2d
12 changed files with 165 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files
config/config.py
+1 -1
View File
@@ -118,7 +118,7 @@ def get_default_level_2_daily_config() -> tuple[dict, dict, dict]:
regression_models = ["Lasso", "KNN", "RF"]
regression_ensemble_model = 'KNN'
classification_models = ['SVC', 'LDA', 'KNN', 'CART', 'NB', 'AB', 'RF', 'StaticMom']
classification_models = ['SVC', 'LDA', 'KNN', 'CART', 'NB', 'AB', 'RF', 'XGB_two_class', 'LGBM', 'StaticMom']
classification_ensemble_model = 'LDA'
model_config = dict(