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)