load_yago3_10

ampligraph.datasets.load_yago3_10(data_home=None)

Load the YAGO3-10 dataset

The dataset is presented in [MBS13]. 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_yago3_10
>>> X = load_yago3_10()
>>> X["valid"][0]
array(['Mikheil_Khutsishvili', 'playsFor', 'FC_Merani_Tbilisi'], dtype=object)