TensorFlow学习笔记之--[tf.app.flags使用方法]

很多时候在运行python代码的时候我们需要从外部定义参数,从而避免每次都需要改动代码。所以一般我们都会使用 argparse 这个库。其实TensorFlow也提供了这个功能,那就是 tf.app.flags

使用方法很简单

tf.app.flags.DEFINE_boolean("param_name", "default_val", "description")

上面给出的是定义一个bool变量,第一个参数是指参数名,第二个是默认值,第三个是对该变量的描述,如果不想描述可以直接用 ""

除了bool类,我们还可以定义其他的类型数据,如:

  • tf.app.flags.DEFINE_integer
  • tf.app.flags.DEFINE_float
  • tf.app.flags.DEFINE_string

那么如何使用呢?完整示例(假设文件名为test.py)如下:

# coding=utf-8

import tensorflow as tf

flags = tf.app.flags
FLAGS = flags.FLAGS
tf.app.flags.DEFINE_integer('data', 10, "")
tf.app.flags.DEFINE_boolean("istrain", True, "")

def main(_):
	print("{}".format(FLAGS.data))
	print("{}".format(FLAGS.istrain))

if __name__ == '__main__':
    tf.app.run()

5-6行:首先需要定义一个tf.app.flags,然后定义一个FLAGS,它是用来解析传入的参数的。

7-8行:定义了两个变量,分别是整型变量和bool型。

10-12行:注意使用tf.app.flags一般需要使用main函数作为入口,然后再仔细看main函数是需要传参数的(虽然不知道为什么),否则会出现如下报错信息:

Traceback (most recent call last):
  File "d:/Code/AutoML/enas/testfiles/batch_test.py", line 19, in <module>
    tf.app.run()
  File "D:\Continuum\anaconda3\lib\site-packages\tensorflow\python\platform\app.py", line 125, in run
    _sys.exit(main(argv))
TypeError: main() takes 0 positional arguments but 1 was given

所以我们可以传入一个无意义的 _ 来解决这个问题。

下面来看看如何运行这个文件:

python test.py --data 20 

>>> 
20
True

可以看到如果不穿入参数则按默认值处理,否则根据传入的值对变量进行更新。

另外对于bool型变量有个地方需要注意,因为bool只有True和False,所以无论bool变量默认值为True还是False,在变量面前加个no后都取False,,其他类型的没有这个特权,示例如下:

python test.py --nodata --noistrain

>>>
10
False

istrain默认值设为False,运行结果和上面一样。




微信公众号:AutoML机器学习
TensorFlow学习笔记之--[tf.app.flags使用方法]
MARSGGBO♥原创
如有意合作或学术讨论欢迎私戳联系~
邮箱:marsggbo@foxmail.com
2018-12-3



上一篇:使用正则表达式(python)


下一篇:耳机和电脑连接后自动关机