SplittingLazyDataset

class SplittingLazyDataset(*, random_state: None | int | Generator = None, ratios: tuple[float, float] | tuple[float, float, float] | None = None, **kwargs: Unpack[LazyDatasetKwargs])[source]

Bases: LazyDataset, ABC

A dataset that splits.

Initialize the dataset.

Parameters:

random_state (TorchRandomHint)
ratios (TransductiveRatiosHint | None)
kwargs (Unpack[LazyDatasetKwargs])