Tensorflow #1 祖传例子 MNIST 手写识别
Tensorflow自带的那个MNIST任务其实挺好用的,之前使用最简单的方法去做,记得正确率应该是92%附近?
其实那个例子是用来熟悉Tensorflow的,算是一个对Tensorflow的熟悉吧,而最近也在看RNN,并且正好看到了这个RNN的例子,因此特别的读了下源代码,并且提供了相关的注释。
如果有问题请留言就好,目前也不是特别熟悉,难免遇到问题。
这个程序总体来说,还是MNIST任务,如果有疑惑可以看我附上上一篇,或者文章最后我直接复制过来的那些。
正确率大约在96%~97%
from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf
import numpy as np
#在这里做数据加载,还是使用那个MNIST的数据,以one_hot的方式加载数据,记得目录可以改成之前已经下载完成的目录
mnist = input_data.read_data_sets("/tmp/data/", one_hot=True)
'''
MNIST的数据是一个28*28的图像,这里RNN测试,把他看成一行行的序列(28维度(28长的sequence)*28行)
'''
# RNN学习时使用的参数
learning_rate = 0.001
training_iters = 100000
batch_size = 128
display_step = 10
# 神经网络的参数
n_input = 28 # 输入层的n
n_steps = 28 # 28长度
n_hidden = 128