Tensorflow中的高级API:Estimator

本文介绍了如何使用Tensorflow的高级API Estimator实现MNIST手写数字识别。通过导入必要的包,设置日志级别,定义主函数以及estimator对象,详细展示了训练和评估模型的过程。特别是cnn_model_fn()方法,根据不同的模式(训练、测试、预测)返回相应的EstimatorSpec对象,指导模型执行相应操作。
摘要由CSDN通过智能技术生成

一、利用 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() 函数,

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值