load_builtin_dataset

class colibri.data.utils.load_builtin_dataset(name, path, train, download)

Bases:

Load a built-in dataset. :param name: The name of the dataset. :type name: str :param path: The path to save the dataset. :type path: str :param train: Whether to load the training or test set. :type train: bool :param download: Whether to download the dataset if it is :type download: bool :param **kwargs: Additional keyword arguments to pass to the pytorch dataset loader.

Returns:

A dictionary containing the input and output data of the dataset.

Return type:

dict

Raises:

KeyError – If the specified dataset name is not found.

Parameters:

• name (str)

• path (str)

• train (bool)

• download (bool)