advsecurenet.datasets.MNIST package
advsecurenet.datasets.MNIST.mnist_dataset module
- class advsecurenet.datasets.MNIST.mnist_dataset.FashionMNISTDataset(preprocess_config: PreprocessConfig | None = None)
Bases:
MNISTDatasetThe FashionMNISTDataset class that loads the Fashion-MNIST dataset.
- Parameters:
preprocess_config (Optional[PreprocessConfig], optional) – The preprocessing configuration for the Fashion-MNIST dataset. Defaults to None.
- mean
Mean of the Fashion-MNIST dataset.
- Type:
List[float]
- std
Standard deviation of the Fashion-MNIST dataset.
- Type:
List[float]
- input_size
Input size of the Fashion-MNIST images.
- Type:
Tuple[int, int]
- name
Name of the dataset.
- Type:
str
- num_classes
Number of classes in the Fashion-MNIST dataset.
- Type:
int
- get_dataset_class()
Returns the dataset class.
- class advsecurenet.datasets.MNIST.mnist_dataset.MNISTDataset(preprocess_config: PreprocessConfig | None = None)
Bases:
BaseDatasetThe 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.