hyperimpute.plugins.utils.metrics module

MAE(X: ndarray, X_true: ndarray, mask: ndarray) ndarray

Mean Absolute Error (MAE) between imputed variables and ground truth.

Parameters:
  • X – Data with imputed variables.

  • X_true – Ground truth.

  • mask – Missing value mask (missing if True)

Returns:

np.ndarray

Return type:

MAE

RMSE(X: ndarray, X_true: ndarray, mask: ndarray) ndarray

Root Mean Squared Error (MAE) between imputed variables and ground truth

Parameters:
  • X – Data with imputed variables.

  • X_true – Ground truth.

  • mask – Missing value mask (missing if True)

Returns:

np.ndarray

Return type:

RMSE