OGBWikiKG2

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

Bases: OGBLoader[WikiKG2TrainDict, WikiKG2EvalDict]

The OGB WikiKG2 dataset.

See also

https://ogb.stanford.edu/docs/linkprop/#ogbl-wikikg2

Initialize the OGB loader.

Parameters:

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