1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
|
# flake8: noqa
from .karate import KarateClub
from .tu_dataset import TUDataset
from .gnn_benchmark_dataset import GNNBenchmarkDataset
from .planetoid import Planetoid
from .nell import NELL
from .citation_full import CitationFull, CoraFull
from .coauthor import Coauthor
from .amazon import Amazon
from .ppi import PPI
from .reddit import Reddit
from .reddit2 import Reddit2
from .flickr import Flickr
from .yelp import Yelp
from .amazon_products import AmazonProducts
from .qm7 import QM7b
from .qm9 import QM9
from .md17 import MD17
from .zinc import ZINC
from .aqsol import AQSOL
from .molecule_net import MoleculeNet
from .pcqm4m import PCQM4Mv2
from .entities import Entities
from .rel_link_pred_dataset import RelLinkPredDataset
from .ged_dataset import GEDDataset
from .attributed_graph_dataset import AttributedGraphDataset
from .mnist_superpixels import MNISTSuperpixels
from .faust import FAUST
from .dynamic_faust import DynamicFAUST
from .shapenet import ShapeNet
from .modelnet import ModelNet
from .coma import CoMA
from .shrec2016 import SHREC2016
from .tosca import TOSCA
from .pcpnet_dataset import PCPNetDataset
from .s3dis import S3DIS
from .geometry import GeometricShapes
from .bitcoin_otc import BitcoinOTC
from .gdelt_lite import GDELTLite
from .icews import ICEWS18
from .gdelt import GDELT
from .willow_object_class import WILLOWObjectClass
from .pascal import PascalVOCKeypoints
from .pascal_pf import PascalPF
from .snap_dataset import SNAPDataset
from .suite_sparse import SuiteSparseMatrixCollection
from .word_net import WordNet18, WordNet18RR
from .freebase import FB15k_237
from .wikics import WikiCS
from .webkb import WebKB
from .wikipedia_network import WikipediaNetwork
from .heterophilous_graph_dataset import HeterophilousGraphDataset
from .actor import Actor
from .upfd import UPFD
from .github import GitHub
from .facebook import FacebookPagePage
from .lastfm_asia import LastFMAsia
from .deezer_europe import DeezerEurope
from .gemsec import GemsecDeezer
from .twitch import Twitch
from .airports import Airports
from .lrgb import LRGBDataset
from .malnet_tiny import MalNetTiny
from .omdb import OMDB
from .polblogs import PolBlogs
from .email_eu_core import EmailEUCore
from .linkx_dataset import LINKXDataset
from .elliptic import EllipticBitcoinDataset
from .elliptic_temporal import EllipticBitcoinTemporalDataset
from .dgraph import DGraphFin
from .hydro_net import HydroNet
from .airfrans import AirfRANS
from .jodie import JODIEDataset
from .wikidata import Wikidata5M
from .myket import MyketDataset
from .brca_tgca import BrcaTcga
from .neurograph import NeuroGraphDataset
from .web_qsp_dataset import WebQSPDataset
from .dbp15k import DBP15K
from .aminer import AMiner
from .ogb_mag import OGB_MAG
from .dblp import DBLP
from .movie_lens import MovieLens
from .movie_lens_100k import MovieLens100K
from .movie_lens_1m import MovieLens1M
from .imdb import IMDB
from .last_fm import LastFM
from .hgb_dataset import HGBDataset
from .taobao import Taobao
from .igmc_dataset import IGMCDataset
from .amazon_book import AmazonBook
from .hm import HM
from .ose_gvcs import OSE_GVCS
from .rcdd import RCDD
from .opf import OPFDataset
from .cornell import CornellTemporalHyperGraphDataset
from .fake import FakeDataset, FakeHeteroDataset
from .sbm_dataset import StochasticBlockModelDataset
from .sbm_dataset import RandomPartitionGraphDataset
from .mixhop_synthetic_dataset import MixHopSyntheticDataset
from .explainer_dataset import ExplainerDataset
from .infection_dataset import InfectionDataset
from .ba2motif_dataset import BA2MotifDataset
from .ba_multi_shapes import BAMultiShapesDataset
from .ba_shapes import BAShapes
import torch_geometric.datasets.utils
homo_datasets = [
'KarateClub',
'TUDataset',
'GNNBenchmarkDataset',
'Planetoid',
'NELL',
'CitationFull',
'CoraFull',
'Coauthor',
'Amazon',
'PPI',
'Reddit',
'Reddit2',
'Flickr',
'Yelp',
'AmazonProducts',
'QM7b',
'QM9',
'MD17',
'ZINC',
'AQSOL',
'MoleculeNet',
'PCQM4Mv2',
'Entities',
'RelLinkPredDataset',
'GEDDataset',
'AttributedGraphDataset',
'MNISTSuperpixels',
'FAUST',
'DynamicFAUST',
'ShapeNet',
'ModelNet',
'CoMA',
'SHREC2016',
'TOSCA',
'PCPNetDataset',
'S3DIS',
'GeometricShapes',
'BitcoinOTC',
'GDELTLite',
'ICEWS18',
'GDELT',
'WILLOWObjectClass',
'PascalVOCKeypoints',
'PascalPF',
'SNAPDataset',
'SuiteSparseMatrixCollection',
'WordNet18',
'WordNet18RR',
'FB15k_237',
'WikiCS',
'WebKB',
'WikipediaNetwork',
'HeterophilousGraphDataset',
'Actor',
'UPFD',
'GitHub',
'FacebookPagePage',
'LastFMAsia',
'DeezerEurope',
'GemsecDeezer',
'Twitch',
'Airports',
'LRGBDataset',
'MalNetTiny',
'OMDB',
'PolBlogs',
'EmailEUCore',
'LINKXDataset',
'EllipticBitcoinDataset',
'EllipticBitcoinTemporalDataset',
'DGraphFin',
'HydroNet',
'AirfRANS',
'JODIEDataset',
'Wikidata5M',
'MyketDataset',
'BrcaTcga',
'NeuroGraphDataset',
'WebQSPDataset',
]
hetero_datasets = [
'DBP15K',
'AMiner',
'OGB_MAG',
'DBLP',
'MovieLens',
'MovieLens100K',
'MovieLens1M',
'IMDB',
'LastFM',
'HGBDataset',
'Taobao',
'IGMCDataset',
'AmazonBook',
'HM',
'OSE_GVCS',
'RCDD',
'OPFDataset',
]
hyper_datasets = [
'CornellTemporalHyperGraphDataset',
]
synthetic_datasets = [
'FakeDataset',
'FakeHeteroDataset',
'StochasticBlockModelDataset',
'RandomPartitionGraphDataset',
'MixHopSyntheticDataset',
'ExplainerDataset',
'InfectionDataset',
'BA2MotifDataset',
'BAMultiShapesDataset',
'BAShapes',
]
__all__ = homo_datasets + hetero_datasets + hyper_datasets + synthetic_datasets
|
