advsecurenet.datasets.svhn package
advsecurenet.datasets.svhn.svhn_dataset module
- class advsecurenet.datasets.svhn.svhn_dataset.SVHNDataset(preprocess_config: PreprocessConfig | None = None)
Bases:
BaseDataset
The MNISTDataset class that loads the MNIST dataset.
- Parameters:
preprocess_config (Optional[PreprocessConfig], optional) – The preprocessing configuration for the MNIST dataset. Defaults to None.
- mean
Mean of the MNIST dataset.
- Type:
List[float]
- std
Standard deviation of the MNIST dataset.
- Type:
List[float]
- input_size
Input size of the MNIST images.
- Type:
Tuple[int, int]
- name
Name of the dataset.
- Type:
str
- num_classes
Number of classes in the MNIST dataset.
- Type:
int
- get_dataset_class()
Returns the dataset class.