correct opt.direction for 2d plots

mlr-org · Aug 4, 2017 · 45fe59e · 45fe59e
1 parent 49501c6
commit 45fe59e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/R/renderExampleRunPlotSingleCrit2d.R b/R/renderExampleRunPlotSingleCrit2d.R
@@ -34,10 +34,9 @@ renderExampleRunPlot2d = function(x, iter,
   critfun = control$infill.crit$fun
   se = (x$learner$predict.type == "se")
 
-  opt.direction = 1
-  if (infill.crit.id %in% c("ei")) {
-    opt.direction = -1
-  }
+  # we need to maximize expected improvement
+  opt.direction = getMBOInfillCritMultiplier(control$infill.crit)
+
   opt.path = as.data.frame(mbo.res$opt.path)
 
   idx.init = which(opt.path$dob == 0)