argparse.ArgumentParser()的使用方法

argparse是python用于解析命令行参数和选项的标准模块。

argparse的使用方法可以简化为四步：

import argparse      导入模块
parser = argparse.ArgumentParser()     创建对象
parser.add_argument()               添加参数
parser.parse_args()                     获取参数

example
 

# 导入模块
import argparse
# 创建对象
parser = argparse.ArgumentParser("cifar")
# 添加变量
parser.add_argument('--data', type=str, default='../data', help='location of the data corpus')
parser.add_argument('--arch_weight_decay', type=float, default=1e-3, help='weight decay for arch encoding')
# 解析参数
args = parser.parse_args()

print(args.data)
print(parser.print_help())

其中：

’--data’ 表示输入参数，命令行没有输入的话，则使用默认参数default。

而当 ‘a’ 前面不加 ‘-’ 的话，必须在命令行输入参数，default不管用了。

解析完成后，可以使用args.***进行参数调用，比如: args.data或者args.arch_weight_decay

help可以对参数进行解释   parser.print_help

输出结果如下

../data
usage: cifar [-h] [--data DATA] [--arch_weight_decay ARCH_WEIGHT_DECAY]

optional arguments:
  -h, --help            show this help message and exit
  --data DATA           location of the data corpus
  --arch_weight_decay ARCH_WEIGHT_DECAY
                        weight decay for arch encoding
None