Tensorflow实现用循环神经网络中的LSTM分类手写数字MNIST数据集

“微信公众号”

这次我们用RNN来进行分类的训练，我们使用的数据集是手写数字MNIST数据集。让RNN从每张图片的第一行像素读到最后一行，然后再进行分类判断。

（1）我们构建的RNN结构图。

（2）最后的训练结果：

（3）最后的测试结果：

（4）完整的代码。

# coding:utf-8
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
tf.set_random_seed(1) # set random seed

# 导入数据
mnist = input_data.read_data_sets("MNIST_data",one_hot=True)

# hyperparameters
lr = 0.