Skip to content

Commit 874f2c0

Browse files
committed
update to handle small runs
1 parent ae71a9e commit 874f2c0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

workflows/nt3_mlrMBO/R/mlrMBO3.R

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,9 +59,9 @@ main_function <- function(max.budget = 110, max.iterations = 10, design.size=10,
5959

6060

6161
surr.rf = makeLearner("regr.randomForest", predict.type = "se")
62-
ctrl = makeMBOControl(n.objectives = 1, propose.points = 20,
62+
ctrl = makeMBOControl(n.objectives = 1, propose.points = min(20, propose.points),
6363
impute.y.fun = function(x, y, opt.path, ...) .Machine$integer.max * 0.1 )
64-
ctrl = setMBOControlInfill(ctrl, crit=makeMBOInfillCritCB(), interleave.random.points=propose.points-20)
64+
ctrl = setMBOControlInfill(ctrl, crit=makeMBOInfillCritCB(), interleave.random.points=min(0,propose.points-20))
6565
ctrl = setMBOControlMultiPoint(ctrl, method = "cb")
6666
ctrl = setMBOControlTermination(ctrl, max.evals = max.budget)
6767
ctrl = setMBOControlTermination(ctrl, iters = max.iterations)

0 commit comments

Comments
 (0)