File tree Expand file tree Collapse file tree 2 files changed +12
-3
lines changed Expand file tree Collapse file tree 2 files changed +12
-3
lines changed Original file line number Diff line number Diff line change 1616from .test_evaluator import *
1717from .util import *
1818
19- WORST_POSSIBLE_RESULT = 2 .0
19+ WORST_POSSIBLE_RESULT = 1 .0
2020
2121
2222class ExecuteTaFuncWithQueue (AbstractTAFunc ):
Original file line number Diff line number Diff line change 1- import functools
21import os
32import time
43import traceback
@@ -508,15 +507,25 @@ def run_smbo(self, max_iters=1000):
508507 model = RandomForestWithInstances (types ,
509508 #instance_features=meta_features_list,
510509 seed = 1 , num_trees = 10 )
510+ rh2EPM = RunHistory2EPM4Cost (num_params = num_params ,
511+ scenario = self .scenario ,
512+ success_states = [StatusType .SUCCESS ,
513+ StatusType .MEMOUT ,
514+ StatusType .TIMEOUT ],
515+ impute_censored_data = False ,
516+ impute_state = None )
511517 smac = SMAC (scenario = self .scenario ,
512518 model = model ,
513519 rng = seed ,
520+ runhistory2epm = rh2EPM ,
514521 tae_runner = ta ,
515522 runhistory = runhistory )
516523 elif self .acquisition_function == 'EIPS' :
517524 rh2EPM = RunHistory2EPM4EIPS (num_params = num_params ,
518525 scenario = self .scenario ,
519- success_states = None ,
526+ success_states = [StatusType .SUCCESS ,
527+ StatusType .MEMOUT ,
528+ StatusType .TIMEOUT ],
520529 impute_censored_data = False ,
521530 impute_state = None )
522531 model = UncorrelatedMultiObjectiveRandomForestWithInstances (
You can’t perform that action at this time.
0 commit comments