tf.app.flags和tf.app.run的使用

tf.app.flags主要用于處理命令行參數的解析工作，其實可以理解為一個封裝好了的argparse包（argparse是一種結構化的數據存儲格式，類似于Json、XML）。

?

我們通過tf.app.flags來調用這個flags.py文件，這樣我們就可以用flags.DEFINE_interger/float()來添加命令行參數，而FLAGS=flags.FLAGS可以實例化這個解析參數的類從對應的命令行參數取出參數。
新建test.py文件，并輸入如下代碼，代碼的功能是創建幾個命令行參數，然后把命令行參數輸出顯示

import tensorflow as tf flags = tf.app.flags flags.DEFINE_string('data_dir', '/tmp/mnist', 'Directory with the MNIST data.') flags.DEFINE_integer('batch_size', 5, 'Batch size.') flags.DEFINE_integer('num_evals', 1000, 'Number of batches to evaluate.') FLAGS = flags.FLAGSprint(FLAGS.data_dir, FLAGS.batch_size, FLAGS.num_evals)

?

·? 在命令行中輸入test.py -h就可以查看幫助信息，也就是Directory with the MNIST data.，Batch size和Number of batches to evaluate這樣的消息。

·? 在命令行中輸入test.py --batchsize 10就可以將batch_size的值修改為10！

tf.app.run()

該函數一般都是出現在這種代碼中：

import tensorflow as tfflags = tf.flagsflags.DEFINE_string('str_name', 'hjd', 'str_vale')flags.DEFINE_integer('int_name', 200, 'int_value')flags.DEFINE_bool('bool_name', False, 'bool_value')flags = flags.FLAGS #必須帶參數，否則：'TypeError: main() takes no arguments (1 given)';#main的參數名隨意定義，無要求 def main(agr_123):print(flags.str_name)print(flags.int_name)print(flags.bool_name)if __name__ == '__main__':tf.app.run()#執行main函數

上述第一行代碼表示如果當前是從其它模塊調用的該模塊程序，則不會運行main函數！而如果就是直接運行的該模塊程序，則會運行main函數。

總結

以上是生活随笔為你收集整理的tf.app.flags和tf.app.run的使用的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。