mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-17 21:08:08 +00:00
Feature: Added sweep functionality (#65)
* feat: Parametricized model selection works now. * feat: Fixed errors. Sweep generates and you can run it, but it gives an error for model.only_columns attribute. * feat: Factored the wandb management, default config managment and the model_dictionary out of the run_pipeline to a seperate file. * fix: Took out prints and fixed the mismatch of ensemble models when classifing. * fix(Models): added StaticMomentum model to the dictionary, hopefully fixed sklearn-ex RandomForestRegressor problem * fix(Dependencies): pin scikit-learn-ex's version, moved map_model_name_to_function to `models` * feat(Sweep): added `run_sweep.py` shortcut * feat(Pipeline): skip training a meta model if array is empty Co-authored-by: Mark Aron Szulyovszky <mark.szulyovszky@gmail.com>
This commit is contained in:
co-authored by
Mark Aron Szulyovszky
parent
d3d7184ea4
commit
1c1b8b2e54
+33
-12
@@ -1,28 +1,49 @@
|
||||
program: rnn_sweep.py
|
||||
method: bayes
|
||||
project: integer-sequence
|
||||
program: run_pipeline.py
|
||||
method: grid
|
||||
project: price-forecasting
|
||||
name: Finding best hyperparameters for price prediction
|
||||
early_terminate:
|
||||
type: hyperband
|
||||
min_iter: 2000
|
||||
metric:
|
||||
goal: maximize
|
||||
name: sharpe
|
||||
# early_terminate:
|
||||
# type: hyperband
|
||||
# min_iter: 2000
|
||||
# metric:
|
||||
# goal: maximize
|
||||
# name: sharpe
|
||||
parameters:
|
||||
path : 'data/'
|
||||
path :
|
||||
value: 'data/'
|
||||
sliding_window_size:
|
||||
values: [50, 90, 130, 160, 180, 280, 380, 500]
|
||||
distribution: categorical
|
||||
retrain_every:
|
||||
values: [7, 14, 30, 60, 100]
|
||||
distribution: categorical
|
||||
scaler:
|
||||
values: ['minmax', 'normalize', 'minmax', 'standardize', 'none']
|
||||
distribution: categorical
|
||||
include_original_data_in_ensemble:
|
||||
values: [True, False]
|
||||
value: True
|
||||
method:
|
||||
values: ['classification', 'regression']
|
||||
value: 'classification'
|
||||
forecasting_horizon:
|
||||
values: [1,2,3,4,5,6,7,8,9,10]
|
||||
distribution: categorical
|
||||
load_other_assets:
|
||||
value: False
|
||||
log_returns:
|
||||
value: True
|
||||
own_features:
|
||||
value: []
|
||||
other_features:
|
||||
value: []
|
||||
index_column:
|
||||
value: 'int'
|
||||
level_1_models:
|
||||
value: ["LR", "LDA", "KNN", "CART", "NB", "AB", "RF"]
|
||||
distribution: constant
|
||||
level_2_models:
|
||||
value: ['Ensemble_CART']
|
||||
distribution: constant
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user