CKG

class CKG(random_state: None | int | Generator = 0, **kwargs)

Bases: TabbedDataset

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

Initialize the CKG dataset from [santos2020].

Parameters:

• random_state (None | int | Generator) – The random seed to use in splitting the dataset. Defaults to 0.

• kwargs – keyword arguments passed to pykeen.datasets.base.TabbedDataset.