mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-05 07:07:46 +00:00
feat(Sweep): separated level-1 and level-2 sweep configs, skip assets with too few samples to train on, simplified model mapping (#84)
* feat: Added ensemble models to sweep and configured naming convention. * fix: Default value was misconfigured. * feat(Sweep): separated level-1 and level-2 sweep configs, skip assets with too few samples to train on, simplified model mapping * fix(Sweep): syntax error * chore(Sweep): set sweep names accordingly * fix(Sweep): set sliding window * fix(Sweep): adjusted sweep config * fix(Sweep): removed invalid feature extractor preset Co-authored-by: Mark Aron Szulyovszky <mark.szulyovszky@gmail.com>
This commit is contained in:
+1
-8
@@ -35,15 +35,8 @@ model_map = {
|
||||
AB= SKLearnModel(AdaBoostClassifier(n_estimators=15)),
|
||||
RF= SKLearnModel(RandomForestClassifier(n_jobs=-1, max_depth=20, random_state=1)),
|
||||
StaticMom= StaticMomentumModel(allow_short=True),
|
||||
),
|
||||
"classification_ensemble_models": dict(
|
||||
Ensemble_CART = SKLearnModel(DecisionTreeClassifier()),
|
||||
Ensemble_Average = StaticAverageModel(),
|
||||
),
|
||||
"regression_ensemble_models": dict(
|
||||
Ensemble_Ridge = SKLearnModel(Ridge(alpha=0.1)),
|
||||
Ensemble_Average = StaticAverageModel(),
|
||||
)
|
||||
}
|
||||
|
||||
model_names_classification = list(model_map["classification_models"].keys())
|
||||
@@ -52,7 +45,7 @@ model_names_regression = list(model_map["regression_models"].keys())
|
||||
|
||||
def map_model_name_to_function(model_config:dict, method:str) -> dict:
|
||||
for level in ['level_1_models', 'level_2_models']:
|
||||
model_category = method + '_models' if level=='level_1_models' else method + '_ensemble_models'
|
||||
model_category = method + '_models'
|
||||
model_config[level] = [(model_name, model_map[model_category][model_name]) for model_name in model_config[level]]
|
||||
|
||||
return model_config
|
||||
Reference in New Issue
Block a user