Skip to content

Commit 88c3251

Browse files
committed
updating mlrmbo to handle small runs
1 parent 874f2c0 commit 88c3251

File tree

3 files changed

+12
-3
lines changed

3 files changed

+12
-3
lines changed

workflows/nt3_mlrMBO/R/mlrMBO3.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ main_function <- function(max.budget = 110, max.iterations = 10, design.size=10,
6161
surr.rf = makeLearner("regr.randomForest", predict.type = "se")
6262
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=min(0,propose.points-20))
64+
ctrl = setMBOControlInfill(ctrl, crit=makeMBOInfillCritCB(), interleave.random.points=max(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)

workflows/p2b1_mlrMBO/R/mlrMBO3.R

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,10 +57,15 @@ simple.obj.fun = function(x){}
5757

5858
main_function <- function(max.budget = 110, max.iterations = 10, design.size=10, propose.points=10){
5959

60+
6061
surr.rf = makeLearner("regr.randomForest", predict.type = "se")
61-
ctrl = makeMBOControl(n.objectives = 1, propose.points = propose.points)
62+
ctrl = makeMBOControl(n.objectives = 1, propose.points = min(20, propose.points),
63+
impute.y.fun = function(x, y, opt.path, ...) .Machine$integer.max * 0.1 )
64+
ctrl = setMBOControlInfill(ctrl, crit=makeMBOInfillCritCB(), interleave.random.points=max(0,propose.points-20))
65+
ctrl = setMBOControlMultiPoint(ctrl, method = "cb")
6266
ctrl = setMBOControlTermination(ctrl, max.evals = max.budget)
6367
ctrl = setMBOControlTermination(ctrl, iters = max.iterations)
68+
6469
# ctrl = setMBOControlInfill(ctrl, crit =makeMBOInfillCritCB(), opt.focussearch.points = 500)
6570
design = generateDesign(n = design.size, par.set = getParamSet(obj.fun))
6671
configureMlr(show.info = FALSE, show.learner.output = FALSE, on.learner.warning = "quiet")

workflows/p3b1_mlrMBO/R/mlrMBO3.R

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,13 @@ simple.obj.fun = function(x){}
5858
main_function <- function(max.budget = 110, max.iterations = 10, design.size=10, propose.points=10){
5959

6060
surr.rf = makeLearner("regr.randomForest", predict.type = "se")
61-
ctrl = makeMBOControl(n.objectives = 1, propose.points = propose.points)
61+
ctrl = makeMBOControl(n.objectives = 1, propose.points = min(20, propose.points),
62+
impute.y.fun = function(x, y, opt.path, ...) .Machine$integer.max * 0.1 )
63+
ctrl = setMBOControlInfill(ctrl, crit=makeMBOInfillCritCB(), interleave.random.points=max(0,propose.points-20))
64+
ctrl = setMBOControlMultiPoint(ctrl, method = "cb")
6265
ctrl = setMBOControlTermination(ctrl, max.evals = max.budget)
6366
ctrl = setMBOControlTermination(ctrl, iters = max.iterations)
67+
6468
# ctrl = setMBOControlInfill(ctrl, crit =makeMBOInfillCritCB(), opt.focussearch.points = 500)
6569
design = generateDesign(n = design.size, par.set = getParamSet(obj.fun))
6670
configureMlr(show.info = FALSE, show.learner.output = FALSE, on.learner.warning = "quiet")

0 commit comments

Comments
 (0)