4e295f0293da827719acdfa9b36875871e5e08b7,examples/bayesian_optimization.py,,,#,10

Before Change

        if n_query == 0:
            plt.ylabel("Expected improvement")
        plt.plot(X, 5*utility, c="r")
        plt.ylim([-0.1, 1])

        // query
        query_idx, query_inst = optimizer.query(X)
        optimizer.teach(X[query_idx].reshape(1, -1), y[query_idx].reshape(1, -1))

After Change

max_PI_tr = partial(max_PI, tradeoff=tr)

acquisitions = zip(
    [PI_tr, EI, UCB],
    [max_PI_tr, max_EI, max_UCB],
)

Instances