nip.image_classification.data.MnistDatasetWrapper#

class nip.image_classification.data.MnistDatasetWrapper(root: str | Path, train: bool = True, transform: Callable | None = None, target_transform: Callable | None = None, download: bool = False)[source]#

The MNIST dataset wrapper.

Attributes

`binarification_seed`
`class_to_idx`
`classes`
`height`
`mirrors`
`num_channels`
`processed_folder`
`raw_folder`
`resources`
`selected_classes`
`test_data`
`test_file`
`test_labels`
`train_data`
`train_labels`
`training_file`
`width`

Methods

nip.image_classification.data.MnistDatasetWrapper

Contents

nip.image_classification.data.MnistDatasetWrapper#