PrimeKG

class PrimeKG(random_state: None | int | Generator = 0, **kwargs)[source]

Bases: SingleTabbedDataset

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

Initialize the PrimeKG dataset from [chandak2022].

Parameters:

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

  • kwargs – keyword arguments passed to SingleTabbedDataset.