MIA-GCL / MERIT / run_citeseer.py
run_citeseer.py
Raw
import os
import copy

opt = dict()
opt['device'] = 'cuda:0'
opt['seed'] = '2021'
opt['data'] = 'citeseer'
opt['epochs'] = 500
opt['lr'] = 3e-4
opt['input_dim'] = 3703
opt['momentum'] = 0.8
opt['alpha'] = 0.05
opt['beta'] = 0.6
opt['drop_edge'] = 0.4
opt['drop_feat1'] = 0.4
opt['drop_feat2'] = 0.4


def command(opt):
    script = 'python train.py'
    for opt, val in opt.items():
        script += ' --' + opt + ' ' + str(val)
    return script


def run(opt):
    opt_ = copy.deepcopy(opt)
    os.system(command(opt_))


if __name__ == '__main__':
    run(opt)