CKG¶

class CKG(create_inverse_triples=False, random_state=0, **kwargs)[source]¶

Bases: pykeen.datasets.base.TabbedDataset

The Clinical Knowledge Graph (CKG) dataset from [santos2020].

Initialize the CKG dataset from [santos2020].

Parameters

create_inverse_triples (bool) – Should inverse triples be created? Defaults to false.
random_state (Union[None, int, Generator]) – The random seed to use in splitting the dataset. Defaults to 0.
kwargs – keyword arguments passed to pykeen.datasets.base.TabbedDataset.