parser.add_argument('--pooling', default='avg', type=str) parser.add_argument('--gcn_dim', default=150, type=int) parser.add_argument('--num_layers', default=2, type=int, help='num of AGGCN layer blocks') parser.add_argument('--gcn_dropout', default=0.5, type=float) parser.add_argument('--map_file', default='maps.pkl', type=str) parser.add_argument('--result_path', default='result', type=str) parser.add_argument('--emb_file', default='100.utf8', type=str) parser.add_argument('--train_file', default=os.path.join("data_doc", "train")) parser.add_argument('--dev_file', default=os.path.join("data_doc", "dev")) parser.add_argument('--test_file', default=os.path.join("data_doc", "test")) parser.add_argument('--log_name', default='test', type=str) parser.add_argument('--seed', default=1023, type=int) args = parser.parse_args() log = Log(args.log_name + ".log") log_handler = log.getLog() log_handler.info("\nArgs: ") for arg in vars(args): log_handler.info("{}: {}".format(arg, getattr(args, arg))) log_handler.info("\n") main()