INNER CODE UNIT · Python
ImputationKernel
AnotherSamWilson/miceforest · miceforest/imputation_kernel.py:35
class ImputationKernel(ImputedData):
"""
Creates a kernel dataset. This dataset can perform MICE on itself,
and impute new data from models obtained during MICE.
Parameters
----------
data : pandas.DataFrame.
The data to be imputed.
variable_schema : None or List[str] or Dict[str, str], default=None
Specifies the feature - target relationships used to train models.
This parameter also controls which models are built. Models can be built
even if a variable contains no missing values, or is not being imputed.
- If :code:`None`, all columns with missing values will have models trained, and all
columns will be used as features in these models.
- If :code:`List[str]`, all columns in data are used to impute the variables in the list
- If :code:`Dict[str, str]` the values will be used to impute the keys.