tabensemb.data.datamodule.DataModule.set_feature_names#

method

DataModule.set_feature_names(all_feature_names: List[str])[source]#

Set feature names to a subset of current features (i.e. self.all_feature_names) and reload the data.

Parameters:
all_feature_names

A subset of current features.