load_fb15k_237

ampligraph.datasets.load_fb15k_237(data_home=None)

Load the FB15k-237 dataset

FB15k-237 is a reduced version of FB15k. It was first proposed by [TCP+15].

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_fb15k_237
>>> X = load_fb15k_237()
>>> X["train"][2]
array(['/m/07s9rl0', '/media_common/netflix_genre/titles', '/m/0170z3'],
  dtype=object)