Example #1
0
    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()