kd.data.py.SliceDataset#

class kauldron.data.py.SliceDataset( start: int | None = <object object>, stop: int | None = <object object>, step: int | None = <object object>, )[source]

Bases: object

Transform which select a subset of the dataset.

Can be useful to debug or train on a subset of the data.

ds = kd.data.py.Tfds(
    name='mnist',
    split='train',
    transforms=[
        kd.data.py.SliceDataset(10),  # Select ds[:10]
    ],
)