PrimeKG

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

Bases: pykeen.datasets.base.SingleTabbedDataset

The Precision Medicine Knowledge Graph (PrimeKG) dataset from [chandak2022].

Initialize the PrimeKG dataset from [chandak2022].

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.SingleTabbedDataset.