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)