ksuit.data.wrappers.dataset_wrappers.repeat_wrapper

Classes

RepeatWrapper

Repeats the wrapped dataset repetitions times.

Module Contents

class ksuit.data.wrappers.dataset_wrappers.repeat_wrapper.RepeatWrapper(dataset, repetitions)

Bases: ksuit.data.base.Subset

Repeats the wrapped dataset repetitions times.

Example

>>> from ksuit.data import ListDataset
>>> dataset = ListDataset([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
>>> len(dataset)
10
>>> repeat_dataset = RepeatWrapper(dataset, repetitions=3)
>>> len(repeat_dataset)
30

Parameters:

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

• repetitions (int) – The number of times to repeat.

Raises:

ValueError – If repetitions is less than 2 or if the dataset is empty. You don’t need to use this wrapper with repetitions < 2.

repetitions