-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_extender_robust_model_experiment.py
36 lines (30 loc) · 1.26 KB
/
run_extender_robust_model_experiment.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import sys
from hyperopt.pyll import scope
from run_experiment import run_experiment
from data_keeper import get_data_keeper
from hyperparameter_search import get_simple_feature_adder_wrapper_params,\
get_feature_selector_params, \
get_model_params, \
get_complex_features_adder_wrapper, \
get_nothing_doing_extender_strategy, \
get_frn_params, \
get_boruta_feature_selector_params, \
get_feature_selector_estimator_params
from common import MAX_EVALS
def get_all_params():
inner_model_params = get_feature_selector_estimator_params()
result_params = get_simple_feature_adder_wrapper_params(
inner_model_params=inner_model_params,
)
return result_params
def run_extender_robust_model(drug):
params = get_all_params()
return run_experiment(
params=params,
experiment_name='extender_robust_model',
drug=drug,
as_indexes=True,
max_evals=MAX_EVALS,
)
if __name__ == '__main__':
run_extender_robust_model(get_data_keeper().get_possible_second_level_drugs()[int(sys.argv[1])])