1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
from kernel.datasets import get_dataset
def print_dataset(dataset):
num_nodes = num_edges = 0
for data in dataset:
num_nodes += data.num_nodes
num_edges += data.num_edges
print('Name', dataset)
print('Graphs', len(dataset))
print('Nodes', num_nodes / len(dataset))
print('Edges', (num_edges // 2) / len(dataset))
print('Features', dataset.num_features)
print('Classes', dataset.num_classes)
print()
for name in ['MUTAG', 'PROTEINS', 'COLLAB', 'IMDB-BINARY', 'REDDIT-BINARY']:
print_dataset(get_dataset(name))
|
