class OGBWikiKG2(cache_root=None, create_inverse_triples=False)[source]

Bases: OGBLoader[WikiKG2TrainDict, WikiKG2EvalDict]

The OGB WikiKG2 dataset.

Initialize the OGB loader.

  • cache_root (Optional[str]) – An optional override for where data should be cached. If not specified, uses default PyKEEN location with pystow.

  • create_inverse_triples (bool) – Should inverse triples be created? Defaults to false.

Attributes Summary


The name of the dataset to download

Attributes Documentation

name: ClassVar[str] = 'ogbl-wikikg2'

The name of the dataset to download