tabensemb.data.datamodule.DataModule._get_indices#

method

DataModule._get_indices(partition: str = 'train') → ndarray[source]#

Get training/validation/testing indices.

Parameters:

partition: “train”, “val”, “test”, or “all”

Returns:

np.ndarray: The indices of the selected partition.