argparse库的使用
在python中,argparse库可以用于在运行.py文件时提供输入参数,这个库的简单使用代码如下,假设文件名为parser.py:
# -*- coding:utf-8 -*-
import argparse
if "__main__" == __name__:
parser = argparse.ArgumentParser()
parser.add_argument("--dataDir", type=str, default="./mnist")
parser.add_argument("--isTraining", type=int, default=0)
parser.add_argument("--sampleCount", type=int, default=5)
flags = parser.parse_args()
print flags.dataDir
print flags.isTraining
print flags.sampleCount
在运行该文件时,可以使用如下命令
python parser.py --dataDir="./Cifar10" --isTraining=1 --sampleCount=10
值得注意的是,当前的argparse并不支持关于bool的解析,本人采用的方法是使用int类型代替