tabensemb.data.datamodule.DataModule.generate_subset#

method

DataModule.generate_subset(dataset: Dataset) Tuple[Subset, Subset, Subset][source]#

Split the dataset into training, validation and testing subsets.

Parameters:
dataset

A torch.utils.data.Dataset instance.

Returns:
torch.utils.data.Subset

Training, validation and testing subsets.