torch_geometric.datasets.AttributedGraphDataset

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

Bases: InMemoryDataset

来自“Scaling Attributed Network Embedding to Massive Graphs”论文的各种属性图数据集。

Parameters:

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

  • name (str) – 数据集的名称 ("Wiki", "Cora", "CiteSeer", "PubMed", "BlogCatalog", "PPI", "Flickr", "Facebook", "Twitter", "TWeibo", "MAG").

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

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

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

统计:

名称

#节点

#edges

#特性

#classes

维基

2,405

17,981

4,973

17

Cora

2,708

5,429

1,433

7

CiteSeer

3,312

4,715

3,703

6

PubMed

19,717

44,338

500

3

博客目录

5,196

343,486

8,189

6

PPI

56,944

1,612,348

50

121

Flickr

7,575

479,476

12,047

9

Facebook

4,039

88,234

1,283

193

TWeibo

2,320,895

9,840,066

1,657

8

MAG

59,249,719

978,147,253

2,000

100