smt_optim.acquisition_strategies package#

class smt_optim.acquisition_strategies.AcquisitionStrategy[source]#

Bases: ABC

abstractmethod get_infill(state) → list[ndarray][source]#

abstractmethod validate_config(state) → None[source]#

class smt_optim.acquisition_strategies.MFSEGO(state: State, **kwargs)[source]#

Bases: AcquisitionStrategy

build_scipy_constraints(acq_context: State) → list[dict][source]#

build_scipy_objective(acq_context: State) → Callable[source]#

compute_all_s2_red_norm(x_pred: ndarray, costs: list[float], surrogates: list) → list[ndarray][source]#

compute_norm_sigma2_red(x_pred: ndarray, norm_costs2: list[float], surrogate) → ndarray[source]#

compute_norm_squared_cost(costs: list[float]) → ndarray[source]#

compute_sigma2_red(x_pred: ndarray, surrogate) → ndarray[source]#

get_fidelity(next_x: ndarray, state: State) → int[source]#

get_infill(acq_context: State) → list[ndarray][source]#

select_fidelity_level(x_pred: ndarray, costs: list[float], all_surrogates: list, criterion: str) → ndarray[source]#

validate_config(acq_context: State) → None[source]#

Submodules#

smt_optim.acquisition_strategies.base module#

class smt_optim.acquisition_strategies.base.AcquisitionStrategy[source]#

Bases: ABC

abstractmethod get_infill(state) → list[ndarray][source]#

abstractmethod validate_config(state) → None[source]#

smt_optim.acquisition_strategies.mfsego module#

class smt_optim.acquisition_strategies.mfsego.MFSEGO(state: State, **kwargs)[source]#

Bases: AcquisitionStrategy

build_scipy_constraints(acq_context: State) → list[dict][source]#

build_scipy_objective(acq_context: State) → Callable[source]#

compute_all_s2_red_norm(x_pred: ndarray, costs: list[float], surrogates: list) → list[ndarray][source]#

compute_norm_sigma2_red(x_pred: ndarray, norm_costs2: list[float], surrogate) → ndarray[source]#

compute_norm_squared_cost(costs: list[float]) → ndarray[source]#

compute_sigma2_red(x_pred: ndarray, surrogate) → ndarray[source]#

get_fidelity(next_x: ndarray, state: State) → int[source]#

get_infill(acq_context: State) → list[ndarray][source]#

select_fidelity_level(x_pred: ndarray, costs: list[float], all_surrogates: list, criterion: str) → ndarray[source]#

validate_config(acq_context: State) → None[source]#