-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathdatasets.py
More file actions
21 lines (20 loc) · 957 Bytes
/
datasets.py
File metadata and controls
21 lines (20 loc) · 957 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from torch_geometric.datasets import *
datasets = [
Planetoid(root='/tmp/Cora', name='Cora'),
Planetoid(root='/tmp/CiteSeer', name='CiteSeer'),
Planetoid(root='/tmp/PubMed', name='PubMed'),
Coauthor(root='/tmp/Physics', name='Physics'),
Coauthor(root='/tmp/ComputerScience', name='CS'),
# CitationFull(root='/tmp/CoraFull', name='Cora'),
WebKB(root='/tmp/Winsconsin', name='Wisconsin'),
#Reddit2(root='/tmp/Reddit'),
# WikiCS(root='/tmp/WikiCS')
Twitch(root='/tmp/TwitchDE', name='DE'),
Twitch(root='/tmp/TwitchPT', name='PT'),
Twitch(root='/tmp/TwitchES', name='ES'),
Amazon(root='/tmp/AmazonPhoto', name='Photo'),
Amazon(root='/tmp/AmazonComputers', name='Computers'),
HeterophilousGraphDataset(root='/tmp/RomanEmpire', name='Roman-empire'),
HeterophilousGraphDataset(root='/tmp/AmazonRatings', name='Amazon-ratings'),
HeterophilousGraphDataset(root='/tmp/Tolokers', name='Tolokers')
]