Ejemplo n.º 1
0
def test_bohb(cs, id):
    bohb = BOHB(cs, train, maximal_iter, num_iter=iter_num, p=0.2, n_workers=n_work)
    bohb.method_name = "BOHB-fcnet-%d" % id
    bohb.runtime_limit = runtime_limit
    bohb.run()
    bohb.plot_statistics(method="BOHB-fcnet-%d" % id)
    print(bohb.get_incumbent(5))
    return bohb.get_incumbent(5)
Ejemplo n.º 2
0
def test_bohb(cs, id):
    model = BOHB(cs,
                 train,
                 maximal_iter,
                 num_iter=iter_num,
                 p=0.2,
                 n_workers=n_work)
    model.method_name = "BOHB-xgb-%d" % id
    model.runtime_limit = runtime_limit
    model.restart_needed = True
    model.run()
    result = model.get_incumbent(5)
    print(model.incumbent_configs)
    print(result)
    return result