torch_geometric.datasets.DBP15K

class DBP15K(root: str, pair: str, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, force_reload: bool = False)[source]

Bases: InMemoryDataset

DBP15K数据集来自 “通过联合属性保留嵌入进行跨语言实体对齐”论文，其中中文、日文和 法文版本的DBpedia与其英文版本进行了链接。 节点特征由预训练和对齐的单语词 嵌入提供，这些嵌入来自“通过图匹配神经网络进行跨语言知识图谱对齐”论文。

Parameters:

  • root (str) – Root directory where the dataset should be saved.

  • pair (str) – 语言对 ("en_zh", "en_fr", "en_ja", "zh_en", "fr_en", "ja_en").

  • transform (callable, optional) – A function/transform that takes in an torch_geometric.data.Data object and returns a transformed version. The data object will be transformed before every access. (default: None)

  • pre_transform (callable, optional) – A function/transform that takes in an torch_geometric.data.Data object and returns a transformed version. The data object will be transformed before being saved to disk. (default: None)

  • force_reload (bool, optional) – Whether to re-process the dataset. (default: False)