The loop API is the most lightweight way to do optimization in Ax. The user makes one call to
optimize, which performs all of the optimization under the hood and returns the optimized parameters.
For more customizability of the optimization procedure, consider the Service or Developer API.
import numpy as np from ax.plot.contour import plot_contour from ax.plot.trace import optimization_trace_single_method from ax.service.managed_loop import optimize from ax.metrics.branin import branin from ax.utils.measurement.synthetic_functions import hartmann6 from ax.utils.notebook.plotting import render, init_notebook_plotting init_notebook_plotting()
[INFO 04-26 20:09:10] ax.utils.notebook.plotting: Injecting Plotly library into cell. Do not overwrite or delete cell.