tabensemb.data.datamodule.DataModule.set_data_splitter#

method

DataModule.set_data_splitter(config: str | Tuple[str, Dict], ratio: List[float] | ndarray | None = None)[source]#

Set the data splitter. The specified splitter should be registered in tabensemb.data.datasplitter.splitter_mapping.

Parameters:

config: The name of a data splitter or a tuple providing the name and kwargs of the data splitter.
ratio: The ratio of training, validation, and testing sets. For example, [0.6, 0.2, 0.2].