argparse的用法解析
argparse是python中常用的模块,在深度学习的相关代码中,经常用该模块来组织参数,可以增强代码的可读性。
使用argpase模块基本上需要以下几个步骤:
import argparse #导入argparse模块
parser = argparse.ArgumentParser() #定义一个类对象
parser.add_argument('--zhengshu', type=int, help='', default=10086) #添加参数,zhengshu时引用参数时的参数名,type是参数类型,help可写可不写,是一些对参数的说明。default是参数的默认值。
parser.add_argument('--zifuchuan', type=str, help='', default='zhangsan')
args = parser.parse_args()
data1 = args.zhengshu
data2 = args.zifuchuan
print(data1)
print(data2)
输出结果是:
注意,在本文中,我们在添加数据时,使用了default参数,这样就不需要在命令行中输入相关的参数了。如果不指定默认参数,需要在命令行中输入专门的参数才可运行。