smac.epm.base_imputor

Classes

BaseImputor()

Abstract implementation of the Imputation API.

class smac.epm.base_imputor.BaseImputor[source]

Bases: object

Abstract implementation of the Imputation API.

abstract impute(censored_X, censored_y, uncensored_X, uncensored_y)[source]

Imputes censored runs and returns new y values.

Parameters
  • censored_X (np.ndarray [N, M]) – Feature array of all censored runs.

  • censored_y (np.ndarray [N, 1]) – Target values for all runs censored runs.

  • uncensored_X (np.ndarray [N, M]) – Feature array of all non-censored runs.

  • uncensored_y (np.ndarray [N, 1]) – Target values for all non-censored runs.

Returns

imputed_y – Same shape as censored_y [N, 1]

Return type

np.ndarray