“微信公众号”
这次我们用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