1_7b_eval_DGL_batched_GIN.py

#!/usr/bin/env python3
import os
import warnings
warnings.filterwarnings("ignore")

hidden = 		[64] 
num_layers = 	[1]
partitions = 	[1500] 

dataset = [
    	('Proteins'             	 , 29     , 2) ,   
		( 'artist'                 	 , 100	  , 12),
		( 'soc-BlogCatalog'	     	 , 128	  , 39),    
]

os.system("touch res_DGL_batched_GIN.log")

for n_Layer in num_layers:
	for hid in hidden:
		for data, d, c in dataset:
			for p in partitions:
				command = "python batched_gin_dgl.py \
        					--gpu 0 \
							--dataset {} \
           					--dim {} \
                            --n-hidden {} \
                            --n-classes {} \
							--psize {}\
							--regular \
							--run_GIN >> res_DGL_batched_GIN.log".\
							format(data, d, hid, c, p)		
				os.system(command)
				print()

os.system("python batched_gin_dgl.py --gpu 0 --dataset ppi --regular --run_GIN >> res_DGL_batched_GIN.log")
print()
os.system("python batched_gin_dgl.py --gpu 0 --dataset ogbn-arxiv --regular --run_GIN >> res_DGL_batched_GIN.log")
print()
os.system("python batched_gin_dgl.py --gpu 0 --dataset ogbn-products --regular >> res_DGL_batched_GIN.log")
print()
os.system("./parse_time.py res_DGL_batched_GIN.log > res_DGL_batched_GIN.csv")
if not os.path.exists("logs"):
	os.system("mkdir logs/")
os.system("mv *.log logs/")