首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

机器学习:运行TensorFlow中的Hello World

正如TensorFlow文档中所讲:当我们开始学习编程的时候,第一件事往往是学习打印"Hello World"。就好比编程入门有Hello World,机器学习入门有MNIST。

MNIST是一个手写数字数据库,它有60000个训练样本集和10000个测试样本集。MNIST数据库官方网址为:http://yann.lecun.com/exdb/mnist/,有四个文件,解压后非独立的图片,而是所有图片(每个图片28*28)按特定格式保存在一起,先行下载好备用。

下来看下在Mac下安装TensorFlow的过程,Mac下自带了python2的环境,但是没有python3,到python官网https://www.python.org/下载3.6 或是3.5的版本安装。python3的版本自带了pip,执行时用pip3命令。疯狂的安装吧。

pip3 install numpy

pip3 install tensorflow

pip3 install keras

pip3 install Pillow

pip3 install IPython

pip3 install matplotlib

一般没什么问题,顺利安装完成。打开python3的控制台。 可以用import tensorflow验证下。

pip3 list 可以列出所有安装过的库:

安装完成后,下载TensorFlow的examples

git clone https://github.com/tensorflow/models.git

完成后进入目录:.../models/official/mnist执行命令开始疯狂的训练模型了。

python3 mnist.py --data_dir=./data --export_dir /tmp/mnist_saved_model

data_dir存放的是之前下载好的文件,export_dir用于导出训练完成后生成的模型。

漫长的等待,生成模型。

然后根据教程愉快的打开TensorBoard,居然没有此命令。

最终还是在site-packages下找到了,

python3 /Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packages/tensorboard/main.py --logdir=/tmp/mnist/mnist_model/

顺利启动TensorBoard,在浏览器中输入http://localhost:6006,打开界面,不明觉厉:

最后用训练好的模型识别下5,3两个数字。我这里挪动了导出的模型文件:

以悲剧结尾,居然识别完是 3, 3

参考官方:

https://github.com/tensorflow/models/tree/master/official/mnist

逆水行舟,继续努力。

  • 发表于:
  • 原文链接http://kuaibao.qq.com/s/20180104G0MX1A00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码

添加站长 进交流群

领取专属 10元无门槛券

私享最新 技术干货

扫码加入开发者社群
领券