tabensemb.data.datamodule.DataModule.get_tabular_dataset#

method

DataModule.get_tabular_dataset(transformed: bool = False) Tuple[DataFrame, List, List, List][source]#

Get the tabular dataset loaded in the DataModule.

Parameters:
transformed

Whether to return the scaled data or not.

Returns:
pd.DataFrame

The tabular dataset.

List

The continuous feature names in the dataset.

List

The categorical feature names in the dataset.

List

The target names.