tabensemb.model.TorchModel._generate_dataset

method

TorchModel._generate_dataset(datamodule: DataModule, model_name: str)

Generate torch.utils.data.Dataset for training.

Parameters:

datamodule

The tabensemb.data.datamodule.DataModule returned by _prepare_custom_datamodule().

model_name

The name of the selected model.

Returns:

torch.utils.data.Dataset

See also

_generate_dataset_from_tensors()