NeuralTSNE.DatasetLoader package
Submodules
NeuralTSNE.DatasetLoader.get_datasets module
- get_dataset(dataset_name: str) Tuple[Dataset, Dataset] | Tuple[None, None][source]View on GitHub
Gets the dataset from the available datasets.
- Parameters:
dataset_name (str) – Name of the dataset.
- Returns:
Tuple containing training and testing datasets or None if the dataset is not available.
- Return type:
Tuple[Dataset, Dataset] | Tuple[None, None]
- get_fashion_mnist() Tuple[Dataset, Dataset][source]View on GitHub
Retrieves the Fashion MNIST dataset from torchvision.
- Returns:
Tuple containing training and testing datasets.
- Return type:
Tuple[Dataset, Dataset]
- get_mnist() Tuple[Dataset, Dataset][source]View on GitHub
Retrieves the MNIST dataset from torchvision.
- Returns:
Tuple containing training and testing datasets.
- Return type:
Tuple[Dataset, Dataset]
- prepare_dataset(dataset_name: str) Tuple[Dataset, Dataset][source]View on GitHub
Loads the dataset from file or creates it if it does not exist. Returns the training and testing datasets.
- Parameters:
dataset_name (str) – Name of the dataset.
- Returns:
Tuple containing training and testing datasets.
- Return type:
Tuple[Dataset, Dataset]