load_wn18

ampligraph.datasets.load_wn18(data_home=None)

Load the WN18 dataset

WN18 is a subset of Wordnet. It was first presented by [BUGD+13]. The dataset 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_wn18
>>> X = load_wn18()
>>> X['test'][:3]
array([['06845599', '_member_of_domain_usage', '03754979'],
       ['00789448', '_verb_group', '01062739'],
       ['10217831', '_hyponym', '10682169']], dtype=object)