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)