File size: 1,130 Bytes
a48216a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import argparse
def get_parser():
parser = argparse.ArgumentParser()
parser.add_argument("--data_dir", default="telecom_data/", type=str,
help="The input data dir. Should contain the training files for the CoNLL-2003 NER task.", )
parser.add_argument("--data_file", default="data_filter.pkl", type=str)
parser.add_argument("--ckpt_dir", default="./PLMs/chinese-roberta-wwm-ext", type=str,
help="The checkpoints dir. Should contain the pretrained model.", )
parser.add_argument("--preprocessor", default="BasePreprocessor", type=str,
help="Name of preprocessor.", )
parser.add_argument("--device", default="cuda:0", type=str)
parser.add_argument("--batch_size", default=128, type=int)
parser.add_argument("--max_epoch", default=100, type=int)
parser.add_argument("--top_k", default=5, type=int)
parser.add_argument("--output_name", default='ELECT_test_output.json', type=str)
return parser
'''
python main_elect_inference.py \
--data_file jicheng_questions.json \
--output_name jicheng_questions_output.json
''' |