使用tensorflow的rnn进行手写数字识别
发布时间
阅读量：
阅读量
采用TensorFlow框架构建RNN模型以识别手写数字，代码中附有详尽的注释说明。此外，本人尝试利用相同代码对多位验证码进行训练，但每次预测所得结果均为固定的一组数字，未能训练出具备实用价值的模型，若有相关经验者欢迎交流探讨。
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
from tensorflow.contrib import rnn
import os
os.environ["TF_CPP_MIN_LOG_LEVEL"] = '3'
# 定义网络
def rnn_network(X, W, b, nsteps, diminput, dimhidden):
X1 = tf.transpose(X, [1, 0, 2])
X2 = tf.reshape(X1, [-1, diminput])
H_1 = tf.matmul(X2, W["h1"]) + b["b1"]
H_1 = tf.split(H_1, nsteps, 0)
# 设置神经单元
lstm_cell = tf.nn.rnn_cell.BasicLST全部评论 (0)
还没有任何评论哟~