Skip to content

Commit 9f8cf73

Browse files
committed
ADD use memout and timeouts when modelling performance
1 parent 33463c7 commit 9f8cf73

File tree

2 files changed

+12
-3
lines changed

2 files changed

+12
-3
lines changed

autosklearn/evaluation/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
from .test_evaluator import *
1717
from .util import *
1818

19-
WORST_POSSIBLE_RESULT = 2.0
19+
WORST_POSSIBLE_RESULT = 1.0
2020

2121

2222
class ExecuteTaFuncWithQueue(AbstractTAFunc):

autosklearn/smbo.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
import functools
21
import os
32
import time
43
import 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(

0 commit comments

Comments
 (0)