tabensemb.data.datamodule.DataModule.get_df#

method

DataModule.get_df(imputed: bool, scaled: bool, cat_transformed: bool) DataFrame[source]#

Get the entire dataframe with certain processing steps.

Parameters:
imputed

Whether continuous and categorical features in the dataframe are imputed.

scaled

Whether continuous features in the dataframe are scaled.

cat_transformed

Whether categorical features in the dataframe are ordinal-encoded.

Returns:
pd.DataFrame