torch_geometric.datasets.Entities

class Entities(root: str, name: str, hetero: bool = False, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, force_reload: bool = False)[source]

Bases: InMemoryDataset

关系实体网络 "AIFB", "MUTAG", "BGS" 和 "AM" 来自 “使用图卷积网络建模关系数据” 论文。 训练和测试分割由节点索引给出。

Parameters:

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

  • name (str) – 数据集的名称 ("AIFB", "MUTAG", "BGS", "AM").

  • hetero (bool, 可选) – 如果设置为 True，将会将数据集保存为 HeteroData 对象。 (默认: False)

  • 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)

统计:

名称

#节点

#edges

#特性

#classes

AIFB

8,285

58,086

0

4

上午

1,666,764

11,976,642

0

11

MUTAG

23,644

148,454

0

2

BGS

333,845

1,832,398

0

2