【tensorflowメモ】tf.app.flags

python実行ファイルにコマンドラインオプションを設定するためのモジュール。

#tf_app_flags.py

import tensorflow as tf

#定義
myflags = tf.app.flags.FLAGS

#tf.app.flags.FLAGS.'型'('引数名', '初期値', '説明')
tf.app.flags.DEFINE_integer("int", 0, "int value")
tf.app.flags.DEFINE_string("str", "test", "string value")
tf.app.flags.DEFINE_boolean("bool", True, "bool value")

def main(argv):
  print(myflags.int, myflags.str, myflags.bool)

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

#実行
$ python tf_app_flags.py --str="Hello, World!" --bool=False

#結果
0 Hello, World! False

tf.app.flagsとtf.flagsを見かけるがどっちでも同じようだ。