get_dataset

get_dataset(*, dataset=None, dataset_kwargs=None, training=None, testing=None, validation=None)

Get a dataset, cached based on the given kwargs.

Parameters

• dataset (Union[None, str, Path, Dataset, Type[Dataset]]) – The name of a dataset, an instance of a dataset, or the class for a dataset.

• dataset_kwargs (Optional[Mapping[str, Any]]) – The keyword arguments, only to be used when a class for a dataset is used for the dataset keyword argument.

• training (Union[None, str, Path, CoreTriplesFactory]) – A triples factory for training triples or a path to a training triples file if dataset=None

• testing (Union[None, str, Path, CoreTriplesFactory]) – A triples factory for testing triples or a path to a testing triples file if dataset=None

• validation (Union[None, str, Path, CoreTriplesFactory]) – A triples factory for validation triples or a path to a validation triples file if dataset=None

Return type

Dataset

Returns

An instantiated dataset

Raises

• ValueError – for incorrect usage of the input of the function

• TypeError – If a type is given for dataset but it’s not a subclass of pykeen.datasets.Dataset