一、利用 Estimator 实现 MNIST 识别
先导入包:
import numpy as np
import tensorflow as tf
写主函数:
if __name__ == "__main__":
tf.logging.set_verbosity(tf.logging.INFO)
tf.app.run()
其中 tf.logging 是Tensorflow中的日志模块,我们这里调用方法 set_verbosity() 设置要打印的信息。自己可以选择打印信息的类型。
_allowed_symbols = [
'DEBUG',
'ERROR',
'FATAL',
'INFO',
'TaskLevelStatusMessage',
'WARN',
'debug',
'error',
'fatal',
'flush',
'get_verbosity',
'info',
'log',
'log_if',
'log_every_n',
'log_first_n',
'set_verbosity',
'vlog',
'warn',
'warning',
]
tf.app.run() 调用 main() 函数,