pe.data.image.cifar10 module

class pe.data.image.cifar10.Cifar10(split='train')

Bases: Data

The CIFAR10 dataset.

__init__(split='train')

Constructor.

Parameters:

split (str, optional) – The split of the dataset. It should be either “train” or “test”, defaults to “train”

Raises:

ValueError – If the split is invalid