ksuit.data.wrappers.dataset_wrappers.shuffle_wrapper

Classes

ShuffleWrapper

Shuffles the dataset, optionally with seed.

Module Contents

class ksuit.data.wrappers.dataset_wrappers.shuffle_wrapper.ShuffleWrapper(dataset, seed=None)

Bases: ksuit.data.base.Subset

Shuffles the dataset, optionally with seed.

Parameters:

  • dataset (ksuit.data.base.Dataset) – The dataset to shuffle.

  • seed (int | None) – The seed for shuffling. Defaults to None.

Raises:

ValueError – If the dataset is not an instance of ksuit.data.Dataset, or if the seed is not an integer or None.

seed = None