[干货|实践] Tensorflow学习 - 使用flags定义命令行参数

/* 版权声明:可以任意转载,转载时请务必标明文章原始出处和作者信息 .*/

faiculty: 机器学习、计算机视觉交流群:451429116  

微信公众号、知乎专栏、简书,请搜索: faiculty

本专栏主要为学习记录,可能综合了网上各类文章或博客,如有侵权请联系本人进行删除。  
QQ:769412850.

非常期待各位同学投稿,一起学习。

学习目的:深度学习神经网络往往有过多的Hyperparameter需要调优,优化算法、学习率、卷积核尺寸等很多参数都需要不断调整,使用命令行参数是非常方便的。有两种实现方式,一是利用python的argparse包,二是调用tensorflow自带的app.flags实现。

利用tf.app.flags组件

tf定义了tf.app.flags,用于接受命令行传递参数,相当于接受argv。首先调用自带的DEFINE_string,DEFINE_boolean DEFINE_integer, DEFINE_float设置不同类型的命令行参数及其默认值。当然,也可以在终端用命令行参数修改这些默认值。

其中参数描述为:

第一个参数第二个参数第三个参数参数名称默认值参数描述

import tensorflow as tf

flags = tf.app.flags
############################
#    hyper parameters      #
############################

# For separate margin loss
flags.DEFINE_float('m_plus', 0.9, 'the parameter of m plus')
flags.DEFINE_float('m_minus', 0.1, 'the parameter of m minus')
flags.DEFINE_float('lambda_val', 0.5, 'down weight of the loss for absent digit classes')
FLAGS = tf.app.flags.FLAGS

def main(_):
    print(FLAGS.m_plus)
    print(FLAGS.m_minus)
    print(lambda_val)
if __name__ == '__main__':
    tf.app.run()  #执行main函数    

执行:

root:~/Tensorflowexample$ python flags_example.py 
0.9
0.1
0.5
root:~/Tensorflowexample$ python flags_example.py
root:~/Tensorflowexample$ python flags_example.py --m_plus 1.0 --m_minus 0.2 --lambad_val 0.9
1.0
0.2
0.5

利用python的argparse包

argparse是python的命令行解析工具,或者说可以在python代码中调用shell的一些命令,从而简化和系统命令之间的交互。tensorflow的一些例子中用argparse来定义一些默认命令,通常是全局变量,也是用作和系统命令之间交互的全局设置。详细介绍参考The Python Standard Library中argparse介绍具体编程时,argparse包的使用方法如下:

1,导入argparse模块

import argparse

2. 创建解析器对象ArgumentParser,可以添加参数

parser=argparse.ArgumentParser(description="This is a example program ")

3. add_argument()方法,用来指定程序需要接受的命令参数

parser.add_argument("--echo",type=str,default='hi',dest='m',help="echo the string you use here")
parser.add_argument("--square",type=int,default= 9 ,dest='n',help="display a square of a given number")
args = parser.parse_args()

定位参数:

parser.add_argument("echo",help="echo the string")

可选参数: 在执行程序的时候,定位参数必选,可选参数可选。

add_argument()常用的参数:

dest:如果提供dest,例如dest=”a”,那么可以通过args.a访问该参数

print(args.m)
print(args.n)

default:设置参数的默认值
action:参数触发的动作
store:保存参数,默认
store_const:保存一个被定义为参数规格一部分的值(常量),而不是一个来自参数解析而来的值。
store_ture/store_false:保存相应的布尔值
append:将值保存在一个列表中。
append_const:将一个定义在参数规格中的值(常量)保存在一个列表中

count:参数出现的次数

parser.add_argument("-v", "--verbosity", action="count", default=0, help="increase output verbosity")

version:打印程序版本信息 type:把从命令行输入的结果转成设置的类型 choice:允许的参数值

parser.add_argument("-v", "--verbosity", type=int, choices=[0, 1, 2], help="increase output verbosity")

help:参数命令的介绍

例程完整代码如下

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--echo",type=str,default='hi',dest='m',help="echo the string you use here")
parser.add_argument("--square",type=int,default= 9 ,dest='n',help="display a square of a given number")
args = parser.parse_args()

print(args.m)
print(args.n)

执行:

root:~/Tensorflowexample$ python argparse_example.py
hi
9
root:~/Tensorflowexample$ python argparse_example.py --echo 'hello' --square 4
hello
4

参考:https://www.jianshu.com/p/b8b09084bd1a

    原文作者:流浪者
    原文地址: https://zhuanlan.zhihu.com/p/33249875
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞