load_builtin_dataset

class colibri.data.utils.load_builtin_dataset(name, path, **kwargs)

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 **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)