给python脚本添加命令行参数主要有三种方式:argparse、Fire工具、sys.argv。
目录
1. argparse
import argparse
parser = argparse.ArgumentParser(description='Options')
parser.add_argument('-folds', default=9, type=int,
help="num of folds") #9折交叉验证
parser.add_argument('-bs', default=128, type=int,
help="batch size")
opt = parser.parse_args()
print(opt.folds)
上面的脚本如果直接在Pycharm中执行的话,会输出默认值default。
如果想传入新的参数值,一种方式是直接修改脚本中相应参数的default,比较暴力;另一种方式是从命令行传入新的参数值,可以直接打开命令行窗口,执行脚本 python a.py -folds=100。
还可以在Pycharm中,设置命令行参数,这样就可以在Pycharm集成环境中运行了,而且脚本使用的是新的参数: