tests/testthat/test_exampleRunMultiobj.R

In berndbischl/mlrMBO: Bayesian Optimization and Model-Based Optimization of Expensive Black-Box Functions

context("exampleRunMultiObj")

test_that("exampleRunMultiObj", {

  doRun = function(method, crit, prop.points, indicator = "sms", trafo.y.fun = NULL) {
    # set nugget effect to small value for num stability in this unit test
    learner = makeLearner("regr.km", predict.type = "se", covtype = "matern3_2", nugget = 0.001)
    control = makeMBOControl(n.objectives = 2L,
      propose.points = prop.points, trafo.y.fun = trafo.y.fun)
    control = setMBOControlTermination(control, iters = 1L)
    if (method == "mspot")
      control = setMBOControlInfill(control, crit = crit, opt = "nsga2", opt.nsga2.popsize = 4L,
        opt.nsga2.generations = 1L)
    else
      control = setMBOControlInfill(control, crit = crit, opt = "focussearch", opt.restarts = 1L,
        opt.focussearch.points = 10L, opt.focussearch.maxit = 1L)
    control = setMBOControlMultiObj(control, method = method, dib.indicator = indicator)

    run = exampleRunMultiObj(testf.zdt1.2d, testd.zdt1.2d, learner = learner, control = control, points.per.dim = 4L,
      nsga2.args = list(popsize = 4L, generations = 2L))
    expect_is(run, "MBOExampleRunMultiObj")
    res = renderExampleRunPlot(run, iter = 1L)
    expect_list(res, min.len = 1L, types = c("ggplot", "list"))
  }

  doRun(method = "parego", crit = crit.ei, prop.points = 1L)
  doRun(method = "parego", crit = crit.ei, prop.points = 2L)
  doRun(method = "parego", crit = crit.ei, prop.points = 2L, trafo.y.fun = trafoLog())
  doRun(method = "parego", crit = crit.cb1, prop.points = 1L)
  doRun(method = "parego", crit = crit.cb1, prop.points = 2L)
  doRun(method = "mspot", crit = crit.ei, prop.points = 1L)
  doRun(method = "mspot", crit = crit.ei, prop.points = 4L)
  doRun(method = "mspot", crit = crit.cb1, prop.points = 1L)
  doRun(method = "mspot", crit = crit.cb1, prop.points = 2L)
  doRun(method = "dib", crit = crit.dib1, prop.points = 1L)
  doRun(method = "dib", crit = crit.dib1, prop.points = 2L)
  doRun(method = "dib", crit = crit.dib1, prop.points = 1L, indicator = "eps")
  doRun(method = "dib", crit = crit.dib1, prop.points = 2L, indicator = "eps")
})