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

refactor(Project): removed regression method (we can still use regression models, but we'll need map them to classification later) (#182)

Browse Source 
* refactor(Project): removed regression method (we can still use regression models, but we'll need map them to classes later)

* fix(Training): removed mistakenly left in `method` parameter
This commit is contained in:
Mark Aron Szulyovszky
2022-01-23 17:15:08 +01:00
committed by GitHub
parent 516c8bcc87
commit 5c4a5b0cf1
12 changed files with 11 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/training_steps.py
+1 -4
View File
@@ -28,7 +28,6 @@ def primary_step(
y = y,
target_returns = target_returns,
models = model_config['primary_models'],
method = data_config['method'],
expanding_window = training_config['expanding_window_primary'],
sliding_window_size = training_config['sliding_window_size_primary'],
retrain_every = training_config['retrain_every'],
@@ -94,9 +93,7 @@ def secondary_step(
X = current_predictions,
y = y,
target_returns = target_returns,
models = [model_config['ensemble_model']],
method = data_config['method'],
expanding_window = False,
models = [model_config['ensemble_model']], expanding_window = False,
sliding_window_size = 1,
retrain_every = training_config['retrain_every'],
from_index = from_index,