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