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

The full citation network datasets from the “Deep Gaussian Embedding of Graphs: Unsupervised Inductive Learning via Ranking” paper. Nodes represent documents and edges represent citation links. Datasets include "Cora", "Cora_ML", "CiteSeer", "DBLP", "PubMed".

Parameters:

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

  • name (str) – The name of the dataset ("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) – Whether the original graph is converted to an undirected one. (default: True)

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

STATS:

Name

#nodes

#edges

#features

#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