ConceptNet

class ConceptNet(create_inverse_triples=False, random_state=0, **kwargs)[source]

Bases: pykeen.datasets.base.SingleTabbedDataset

The ConceptNet dataset from [speer2017].

The dataset is structured into 5 columns (see https://github.com/commonsense/conceptnet5/wiki/Downloads#assertions): edge URL, relation, head, tail, metadata.

speer2017

Robyn Speer, Joshua Chin, and Catherine Havasi. (2017) ConceptNet 5.5: An Open Multilingual Graph of General Knowledge. In proceedings of AAAI 31.

Initialize dataset.

Parameters

  • url – The url where to download the dataset from

  • name – The name of the file. If not given, tries to get the name from the end of the URL

  • cache_root – An optional directory to store the extracted files. Is none is given, the default PyKEEN directory is used. This is defined either by the environment variable PYKEEN_HOME or defaults to ~/.pykeen.

  • eager – Should the data be loaded eagerly? Defaults to false.

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