tabensemb.model.RFE._data_preprocess

method

RFE._data_preprocess(df, derived_data, model_name)

Perform the same preprocessing as in _train_data_preprocess() on a new dataset.

Parameters:

df:

The new tabular dataset that has the same structure as self.trainer.datamodule.X_test

derived_data:

Unstacked data derived from tabensemb.data.datamodule.DataModule.derive_unstacked(). If None, unstacked data will be re-derived.

model_name:

The name of a selected model.

Returns:

Any

The processed data that has the same structure as X_test from _train_data_preprocess().

Notes

The input df is not scaled. To scale the df, run df = datamodule.data_transform(df, scaler_only=True)