load_wn18rr

ampligraph.datasets.load_wn18rr(data_home=None)

Load the WN18RR dataset

The dataset is described in [DMSR18]. It is divided in three splits:
  • train
  • valid
  • test
Returns:splits – The dataset splits: {‘train’: train, ‘valid’: valid, ‘test’: test}. Each split is an ndarray of shape [n, 3].
Return type:dict

Examples

>>> from ampligraph.datasets import load_wn18rr
>>> X = load_wn18rr()
>>> X["valid"][0]
array(['02174461', '_hypernym', '02176268'], dtype=object)