mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-18 13:28:22 +00:00
feat(Models): added debug_future_lookahead, sped up LogisticRegression & DecisionTreeClassifier (#74)
* feat(Models): added `debug_future_lookahead`, sped up LogisticRegression & DecisionTreeClassifier * feat(Training): added ability to train on expanding_window * feat(Models): tuned some hyperparameters, added expanding_window to sweep config, fixed tests * feat(Models): tune parameters of ensemble models * fix(Config): use window size that works with ensembling
This commit is contained in:
@@ -40,6 +40,7 @@ def pipeline(project_name:str, wandb, sweep:bool, model_config:dict, training_co
|
||||
target_returns = target_returns,
|
||||
models = model_config['level_1_models'],
|
||||
method = data_config['method'],
|
||||
expanding_window = training_config['expanding_window'],
|
||||
sliding_window_size = training_config['sliding_window_size'],
|
||||
retrain_every = training_config['retrain_every'],
|
||||
scaler = training_config['scaler'],
|
||||
@@ -64,6 +65,7 @@ def pipeline(project_name:str, wandb, sweep:bool, model_config:dict, training_co
|
||||
target_returns = target_returns,
|
||||
models = model_config['level_2_models'],
|
||||
method = data_config['method'],
|
||||
expanding_window = training_config['expanding_window'],
|
||||
sliding_window_size = training_config['sliding_window_size'],
|
||||
retrain_every = training_config['retrain_every'],
|
||||
scaler = training_config['scaler'],
|
||||
|
||||
Reference in New Issue
Block a user