tabensemb.data.dataimputer.MiceLightgbmImputer._get_impute_features

method

MiceLightgbmImputer._get_impute_features(cont_feature_names: List[str], data: DataFrame) → List[str]

Get continuous feature names that can be imputed, i.e. those not totally missing.

Parameters:

cont_feature_names:

Names of continuous features.

data:

The input tabular dataset.

Returns:

List

Names of continuous features that can be imputed.