argparse模块主要用来为脚本传递命令参数功能,使他们更加灵活。argparse模块作为optparse的替代被添加到Python2.7中,使用方法类似于optparse,但是也加入很多新的特征。
可以说argparse是一个全面的参数处理库。当程序中有多个参数,调用parse_args()时,会先识别参数名称所有带前缀'-'的参数,然后再将剩余的参数按照顺序视为对应位置上的参数。
下面是文本分类任务中的一个伪代码,可以参考:
import argparse
parser = argparse.ArgumentParser(description='Chinese Text Classification')
parser.add_argument('--model', type=str, required=True, help='choose a model: TextCNN, TextRNN, FastText, TextRCNN, TextRNN_Att, DPCNN, Transformer')
parser.add_argument('--embedding', default='pre_trained', type=str, help='random or pre_trained')
parser.add_argument('--word', default=False, type=bool, help='True for word, False for char')
args = parser.parse_args()
if __name__ == '__main__':
if args.embedding == 'random':
embedding = 'random'
model_name = args.model
方法A()
if args.word== 'FastText':
方法B()
else:
方法C()
运行时,直接在命令行敲 python your_code.py -your_后缀名 后缀内容