archive design

Author: be-marc

DESCRIPTION

@@ -93,8 +93,8 @@ Collate:
     'AcqFunctionStochasticEI.R'
     'AcqOptimizer.R'
     'aaa.R'
-    'OptimizerAsyncMbo.R'
     'OptimizerADBO.R'
+    'OptimizerAsyncMbo.R'
     'OptimizerMbo.R'
     'mlr_result_assigners.R'
     'ResultAssigner.R'
@@ -104,8 +104,8 @@ Collate:
     'SurrogateLearner.R'
     'SurrogateLearnerAsync.R'
     'SurrogateLearnerCollection.R'
-    'TunerAsyncMbo.R'
     'TunerADBO.R'
+    'TunerAsyncMbo.R'
     'TunerMbo.R'
     'mlr_loop_functions.R'
     'bayesopt_ego.R'

NAMESPACE

@@ -19,8 +19,8 @@ export(AcqFunctionSmsEgo)
 export(AcqFunctionStochasticCB)
 export(AcqFunctionStochasticEI)
 export(AcqOptimizer)
-export(OptimizerAsyncMbo)
 export(OptimizerADBO)
+export(OptimizerAsyncMbo)
 export(OptimizerMbo)
 export(ResultAssigner)
 export(ResultAssignerArchive)
@@ -29,8 +29,8 @@ export(Surrogate)
 export(SurrogateLearner)
 export(SurrogateLearnerAsync)
 export(SurrogateLearnerCollection)
-export(TunerAsyncMbo)
 export(TunerADBO)
+export(TunerAsyncMbo)
 export(TunerMbo)
 export(acqf)
 export(acqfs)

R/OptimizerAsyncMbo.R

@@ -102,7 +102,10 @@ OptimizerAsyncMbo = R6Class("OptimizerAsyncMbo",
       pv = self$param_set$values
 
       # initial design
-      design = if (is.null(pv$initial_design)) {
+      design = if (inst$archive$n_evals) {
+        lg$debug("Using archive with %s evaluations as initial design", inst$archive$n_evals)
+        NULL
+      } else if (is.null(pv$initial_design)) {
         # generate initial design
         generate_design = switch(pv$design_function,
           "random" = generate_design_random,

man/mlr_optimizers_adbo.Rd

@@ -18,3 +18,30 @@ test_that("async optimizer works in defaults", {
   expect_rush_reset(instance$rush)
 })
 
+test_that("async optimizer works with evaluations in archive", {
+  skip_on_cran()
+  skip_if_not_installed("rush")
+  flush_redis()
+
+  rush::rush_plan(n_workers = 2)
+  instance = oi_async(
+    objective = OBJ_2D,
+    search_space = PS_2D,
+    terminator = trm("evals", n_evals = 10),
+  )
+
+  optimizer = opt("async_random_search")
+  optimizer$optimize(instance)
+
+
+  instance$terminator$param_set$values$n_evals = 30
+
+  optimizer = opt("async_mbo")
+  optimizer$optimize(instance)
+
+  instance$archive
+  expect_data_table(instance$archive$data[is.na(get("acq_cb"))], min.rows = 10)
+  expect_data_table(instance$archive$data[!is.na(get("acq_cb"))], min.rows = 20)
+
+  expect_rush_reset(instance$rush)
+})