python 代码传递参数模块-argparse

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_后缀名 后缀内容

猜你喜欢

转载自blog.csdn.net/qq_36663518/article/details/107690311