OGBWikiKG2

class OGBWikiKG2(cache_root: str | None = None, create_inverse_triples: bool = False)

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 (Path) – 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

name	The name of the dataset to download

Attributes Documentation

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

The name of the dataset to download