tabensemb.data.dataimputer.GainImputer.transform

method

GainImputer.transform(input_data: DataFrame, datamodule: DataModule) → DataFrame

Restore feature names in the datamodule using recorded features, and transform the input tabular data using the fitted imputer. This should be performed on the testing set.

Parameters:

input_data:

A tabular dataset.

datamodule:

A DataModule instance that contains necessary information required by imputers.

Returns:

pd.DataFrame

A transformed tabular dataset.