qolmat.imputations.imputers.ImputerOracle

class qolmat.imputations.imputers.ImputerOracle

Perfect imputer, requires to know real values.

Used as a reference to evaluate imputation metrics.

Parameters

dfpd.DataFrame

Dataframe containing real values.

groups: Tuple[str, …]

List of column names to group by, by default []

__init__() → None

set_solution(df: DataFrame)

Set the true values to be returned by the oracle.

Parameters

dfpd.DataFrame

True dataset with mask

transform(X: DataFrame) → DataFrame

Impute df with corresponding known values.

Parameters

Xpd.DataFrame

dataframe to impute

Returns

pd.DataFrame

dataframe imputed with premasked values