Skip to content

Commit 8ab983f

Browse files
committed
updating mlrMO settings
1 parent c52b923 commit 8ab983f

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

workflows/nt3_mlrMBO/R/mlrMBO3.R

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,12 +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-
impute.y.fun = function(x, y, opt.path, ...) .Machine$integer.max * 0.1 )
62+
ctrl = makeMBOControl(n.objectives = 1, propose.points = 20,
63+
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)
65+
ctrl = setMBOControlMultiPoint(ctrl, method = "cb")
6366
ctrl = setMBOControlTermination(ctrl, max.evals = max.budget)
6467
ctrl = setMBOControlTermination(ctrl, iters = max.iterations)
65-
# ctrl = setMBOControlInfill(ctrl, crit =makeMBOInfillCritCB(), opt.focussearch.points = 500)
68+
6669
design = generateDesign(n = design.size, par.set = getParamSet(obj.fun))
6770
print(design)
6871
configureMlr(show.info = FALSE, show.learner.output = FALSE, on.learner.warning = "quiet")

0 commit comments

Comments
 (0)