LoginSignup
8

More than 5 years have passed since last update.

tensorflowメモ:tf.app.*

Last updated at Posted at 2016-11-16

version: 0.11.0

コマンドラインオプション

簡単なコマンドラインオプションを定義できます.

使い方

app.py
import tensorflow as tf

tf.app.flags.DEFINE_boolean("bool", True, "bool value")
tf.app.flags.DEFINE_integer("int", 0, "int value")
tf.app.flags.DEFINE_string("str", "str", "string value")
tf.app.flags.DEFINE_string("test_str", "test", "test string value")

def main(argv):
  flags = tf.app.flags.FLAGS

  print(flags.bool, flags.int, flags.str, flags.test_str)


if __name__ == '__main__':
    tf.app.run()
% python test.py --help
usage: test.py [-h] [--bool [BOOL]] [--nobool] [--int INT] [--str STR]
               [--test_str TEST_STR]

optional arguments:
  -h, --help           show this help message and exit
  --bool [BOOL]        bool value
  --nobool
  --int INT            int value
  --str STR            string value
  --test_str TEST_STR  test string value

% python test.py --int 1000 --test_str "TEST"
(True, 1000, 'str', 'TEST')

API

値の追加

tf.app.flags.DEFINE_boolean(flag_name, default_value, docstring)

tf.app.flags.DEFINE_integer(flag_name, default_value, docstring)

tf.app.flags.DEFINE_string(flag_name, default_value, docstring)

Args:

  • flag_name: 変数名 & コマンドライン・オプション名
  • default_value: デフォルト値
  • docstring: コマンドライン ヘルプ

値の取得

tf.app.flags.Flag

tf.app.flags.DEFINE_boolean(test_bool, True, "test bool value")
tf.app.flags.Flag.test_bool #-> True

パース

tf.app.run(main=None)

Args:
- main: コマンドラインをエスケープした後に実行するargvを引数とする関数func(argv)
tensorflowがエスケープできなかった残りのコマンドライン引数が渡される.
Noneの場合 __main__.main が実行される

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
8