详情请进入 湖南阳光电子学校 已关注:人 咨询电话:0731-85579057 微信号:yp941688, yp94168
背景
近一直训练模型,进行模型调参比较,但是随着模型参数多起来了,想获得不同参数模型结果也越来越麻烦。看看前辈以及github上开源项目的代码后,感觉这个感觉学到了很多。让你的python程序输入参数更像Linux那样输入,之前使用sys.argv,看到这个之后感觉太low了。这个就是 Python 标准库的一部分:Argparse[1]。那么下面让我们看看他是多么方便。
1 入门操作
由于这个是Python标准库中的一部分,所以就不用安装了,直接使用即可。下面结合文章argparse基本用法[2]和自己的理解书写如下程序:
运行方式:linux系统使用比较多的还是命令,这里我也在终端里运行程序,如下是使用默认参数运行:
无参数运行
有参运行如下:
有参运行
看起来是不是挺有意思的。在上面的代码中,使用argparse.ArgumentParser构建一个参数解析器,其实也就是创建一个相关对象,description描述了这个解析器,在命令行中使用-h(--help)命令就可以看到,如下:
帮助文档
是不是很清晰。
使用这个参数解析器得有一些我们需要的参数,那么就使用parser.add_argument()方法去定义一些。上面的例子就是增加了两个参数:name, job。其中'-n'、'--name'都表示一个参数,而default是运行程序时不输入对应参数值所启用的值。现在相信你对这个有一定的了解了。下面我们再深入一下。
2 挖掘argparse更多功能
当然,上面的内容能够应付日常中一些任务。了解更多的内容,也会让我们事半功倍。下面的内容参考了argparse简要用法总结[3]部分内容,特此感谢。主要介绍parser.add_argument()中的其他参数,default就不再介绍了。
2.1 required:表示这个参数是否一定需要设置
如果设置了required=True,则在实际运行的时候不设置该参数将报错,在上面的程序上添加如下代码:
当不输入值时,则有:
必要参数
2.2 type: 参数类型
命令参数是传给其他功能模型使用,不同的功能模型则需要不同类型的数据输入,可以通过设置type直接将输入的数据转换成对应类型的数据(前提输入的数据格式就满足转换要求),修改need参数的代码如下:
运行查看:
指定参数类型
2.3 choices:参数值只能从几个选项里面选择
有时为了更加人性化提示用户使用,可以提升用户输入哪些值,慢慢地感觉到了Linux的味道了。现在我们修改need参数的代码如下:
使用-h命令查看修改后的样子:
帮助命令
可以看到,need参数的选择项。具体效果,朋友们可以自行试试,这里不费篇幅展示了,以下是输入选择范围外的结果。
选择范围外结果
2.4 dest:修改程序内部获取值方法
在命令行中输入的命令是-ne等形式把参数传入到程序,程序可通过其他属性获取对应参数值,就是dest,修改need参数如下:
运行:
获取真实值
上面还修改打印语句为:
3 总结
argparse是一个全面的参数处理库,功能比较强大,内容也比较多,暂时就介绍这么多吧。更多实际需要的功能,我们就根据需要查看官方文档以及网上的其他资料吧。
参考资料
[1]Argparse: https://docs.python.org/zh-cn/3.7/howto/argparse.html
[2]argparse基本用法: https://blog.csdn.net/yy_diego/article/details/82851661
[3]argparse简要用法总结: vra.github.io/2017/12/02/argparse-usage/
责任编辑:xj
原文标题:让Python程序参数输入更像Linux命令——argparse
文章出处:【微信公众号:自然语言处理爱好者】欢迎添加关注!文章转载请注明出处。
.(编辑:攀枝花电工培训学校)