tabensemb.model.AutoGluon._data_preprocess#

method

AutoGluon._data_preprocess(df, derived_data, model_name)[source]#

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)