-
Notifications
You must be signed in to change notification settings - Fork 14
/
Copy path2_tcgnn_single_kernel.py
executable file
·33 lines (28 loc) · 1.05 KB
/
2_tcgnn_single_kernel.py
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
#!/usr/bin/env python3
import os
os.environ["PYTHONWARNINGS"] = "ignore"
model = 'gcn'
hidden = 16
dataset = [
('citeseer' , 3703 , 6 ),
('cora' , 1433 , 7 ),
('pubmed' , 500 , 3 ),
('ppi' , 50 , 121 ),
('PROTEINS_full' , 29 , 2) ,
('OVCAR-8H' , 66 , 2) ,
('Yeast' , 74 , 2) ,
('DD' , 89 , 2) ,
('YeastH' , 75 , 2) ,
( 'amazon0505' , 96 , 22),
( 'artist' , 100 , 12),
( 'com-amazon' , 96 , 22),
( 'soc-BlogCatalog' , 128 , 39),
( 'amazon0601' , 96 , 22),
]
for data, _, _ in dataset:
command = "python main_tcgnn.py --dataset {} \
--dim {} \
--hidden {} \
--single_kernel" \
.format(data, hidden, hidden)
os.system(command)