tabensemb.data.datamodule.DataModule.extract_original_cat_feature_names

method

DataModule.extract_original_cat_feature_names(all_feature_names: List[str]) → List[str]

Get original categorical features specified in the configuration.

Parameters:

all_feature_names

A list of features that contains some original features in the configuration.

Returns:

List

Names of categorical original features that are both in the configuration and the input list.

See also

extract_original_cont_feature_names(), extract_derived_stacked_feature_names()