torch_geometric.datasets.CitationFull

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

Bases: InMemoryDataset

来自“Deep Gaussian Embedding of Graphs: Unsupervised Inductive Learning via Ranking”论文的完整引用网络数据集。 节点代表文档，边代表引用链接。 数据集包括"Cora", "Cora_ML", "CiteSeer", "DBLP", "PubMed"。

Parameters:

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

  • name (str) – 数据集的名称 ("Cora", "Cora_ML" "CiteSeer", "DBLP", "PubMed").

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

  • to_undirected (bool, optional) – 是否将原始图转换为无向图。（默认值：True）

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

统计:

名称

#节点

#edges

#特性

#classes

Cora

19,793

126,842

8,710

70

Cora_ML

2,995

16,316

2,879

7

CiteSeer

4,230

10,674

602

6

DBLP

17,716

105,734

1,639

4

PubMed

19,717

88,648

500

3