Skip to content

HEBO API Reference

Installation and usage

sb_arch_opt.algo.hebo_interface.get_hebo_optimizer(problem: ArchOptProblemBase, n_init: int, seed: int = None)

Gets the main interface to HEBO. Use the optimize method to run the DOE and infill loops.

sb_arch_opt.algo.hebo_interface.algo.HEBOArchOptInterface(problem: ArchOptProblemBase, n_init: int, seed: int = None)

Interface class to HEBO algorithm.

pop: Population property

optimize(n_infill: int)

Run the optimization loop for n_infill infill points (on top on the initialization points)

ask() -> np.ndarray

Returns n_batch infill points

tell(x: np.ndarray, f: np.ndarray, g: np.ndarray = None)

Updates optimizer with evaluated design points