nip.image_classification.data.MnistDatasetWrapper

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