【TensorFlow学习笔记】tensorflow命令行参数定义

1、argparse

argparse是python的命令行解析库

import argparse

def parse_arguments():
	# 创建解析器 
	parser = argparse.ArgumentParser()
	
	# 第一个是参数名称，第二个参数是类型，第三个参数是默认值，第四个是参数描述 
	parser.add_argument('--logs_base_dir', type=str,help='Directory where to write event logs.', default='~/logs/detailed_graph')

	return parser.parse_args()

def main(args):
	path = args.logs_base_dir

if __name__ == '__main__':
    main(parse_arguments())

2、tf.app.flags.FLAGS-接受命令行传递参数

此方法的底层函数就是Python argparse模块

import tensorflow as tf

# 第一个是参数名称，第二个参数是默认值，第三个是参数描述
tf.app.flags.DEFINE_string('str_name', 'def_v_1', "descrip1")
tf.app.flags.DEFINE_integer('int_name', 10, "descript2")
tf.app.flags.DEFINE_boolean('bool_name', False, "descript3")

FLAGS = tf.app.flags.FLAGS

def main(args):
    print(FLAGS.str_name)
    print(FLAGS.int_name)
    print(FLAGS.bool_name)


if __name__ == '__main__':
    tf.app.run()  # 执行main函数