def test_EI(self):
opt = SimpleBayesianOptimizer({"initial_random_runs": 3})
exp = Experiment("test", {"x": MinMaxNumericParamDef(0, 1)})
for i in range(10):
cand = opt.get_next_candidates(exp)[0]
assert_true(isinstance(cand, Candidate))
cand.result = 2
exp.add_finished(cand)
cands = opt.get_next_candidates(exp, num_candidates=3)
assert_equal(len(cands), 3)
def test_PoI(self):
opt = SimpleBayesianOptimizer({"initial_random_runs": 3, "acquisition": ProbabilityOfImprovement})
assert_true(isinstance(opt.acquisition_function, ProbabilityOfImprovement))
exp = Experiment("test", {"x": MinMaxNumericParamDef(0, 1)})
for i in range(10):
cand = opt.get_next_candidates(exp)[0]
assert_true(isinstance(cand, Candidate))
cand.result = 2
exp.add_finished(cand)
cands = opt.get_next_candidates(exp, num_candidates=3)
assert_equal(len(cands), 3)
