mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-09 09:07:49 +00:00
b6cd6b14fe
* feat(Config): feature extractors are enabled one-by-one with a bool, added previous model to model.fit() * fix(Sweep): removed unused `other_features` parameter that fails sweep * feat(Config): using preset names for defining feature extractors again * fix(Tests): fixed model stub classes
52 lines
1.4 KiB
YAML
52 lines
1.4 KiB
YAML
program: run_sweep.py
|
|
method: bayes
|
|
project: price-forecasting
|
|
name: Finding best hyperparameters for price prediction
|
|
# early_terminate:
|
|
# type: hyperband
|
|
# min_iter: 2000
|
|
metric:
|
|
goal: maximize
|
|
name: sharpe
|
|
parameters:
|
|
path :
|
|
value: 'data/'
|
|
expanding_window:
|
|
values: [True, False]
|
|
distribution: categorical
|
|
sliding_window_size:
|
|
values: [180, 280, 380]
|
|
distribution: categorical
|
|
retrain_every:
|
|
values: [10, 20, 30]
|
|
distribution: categorical
|
|
scaler:
|
|
values: ['minmax', 'normalize', 'minmax', 'standardize', 'none']
|
|
distribution: categorical
|
|
include_original_data_in_ensemble:
|
|
values: [True, False]
|
|
distribution: categorical
|
|
method:
|
|
value: 'classification'
|
|
forecasting_horizon:
|
|
values: [1,2,3,4,5,6,7,8,9,10]
|
|
distribution: categorical
|
|
load_other_assets:
|
|
values: [True, False]
|
|
distribution: categorical
|
|
log_returns:
|
|
values: [True, False]
|
|
distribution: categorical
|
|
index_column:
|
|
value: 'int'
|
|
level_1_models:
|
|
values: [["LR"], ["LDA"], ["KNN"], ["CART"], ["NB"], ["AB"], ["RF"]]
|
|
distribution: categorical
|
|
level_2_models:
|
|
value: []
|
|
distribution: constant
|
|
own_features:
|
|
values: [['only_mom', 'date_days'], [], ['level_1', 'date_days'], ['level_1', 'date_days', 'level_2']]
|
|
distribution: categorical
|
|
other_features:
|
|
value: [] |