- list只能加 a+b
- set 只能减 a-b
- parser 传参数
import argparse
parser = argparse.ArgumentParser(description='Training GNN on Paper-Venue (Journal) classification task')
parser.add_argument('--data_dir', type=str, default='./dataset/oag_output',
help='The address of preprocessed graph.')
parser.add_argument('--model_dir', type=str, default='./model_save',
help='The address for storing the models and optimization results.')
parser.add_argument('--conv_name', type=str, default='hgt',
choices=['hgt', 'gcn', 'gat', 'rgcn', 'han', 'hetgnn'],)
args = parse.parse_args()
open(os.path.join(args.data_dir, 'graph%s.pk' % args.domain), 'rb')