tensorflow1.x循环神经网络,完成手写体识别

import tensorflow as tf
from tensorflow.contrib.layers import fully_connected
from tensorflow.examples.tutorials.mnist import input_data

'''
利用循环神经网络,完成手写体识别
'''
mnist = input_data.read_data_sets("MNIST_data")
# mnist = input_data.read_data_sets(r"MNIST_data", one_hot=True)

# 参数
batch_size = 100 #mini-batch批大小
n_steps = 28 #时间步数(序列长度)
n_inputs = 28 #输入数据长度
n_neurons = 100 #隐藏状态,神经元个数
n_outputs = 10 #输出10分类
learning_rate = 0.001 #学习率
n_epochs = 1 #训练大周期

# 输入输出占位符
X = tf.placeholder(tf.float32, [None, n_steps, n_inputs]) #三维数据(?, 28, 28)
y = tf.placeholder(tf.int64, [None]) # 一维输出,标签0-9
# y = tf.placeholder(tf.float32, shape=[None, 10])

# 模型使用最简单的BasicRNNcell
basic_cell = tf.contrib.rnn.BasicRNNCell(num_units=n_neurons)
outputs, states = tf.nn.dynamic_rnn(basic_cell, X, dtype=tf.float32)  #outputs(?, 28, 100) states(?, 100)

logits = fully_connected(outputs[:,-1], n_outputs, activation_fn=None) #用最后一个时间步的输出

# 代价或损失函数
loss = tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(logits=logits, labels=y))
# loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits_v2(logits=logits, labels=y))
training_op = tf.train.AdamOptimizer(learning_rate=learning_rate).minimize(loss)

# 计算准确率,只有等于y才是对的,其它都错
correct = tf.equal(tf.argmax(logits, 1), y)
accuracy = tf.reduce_mean(tf.cast(correct, tf.float32))
# acc_count = tf.equal(tf.argmax(logits, 1), tf.argmax(y, 1))
# accuracy = tf.reduce_mean(tf.cast(acc_count, dtype=tf.float32))

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    for epoch in range(n_epochs):
        for iteration in range(mnist.train.num_examples // batch_size):  # 55000
            X_batch, y_batch = mnist.train.next_batch(batch_size)
            X_batch = X_batch.reshape((-1, n_steps, n_inputs)) #(?, 28, 28)
            sess.run(training_op, feed_dict={X: X_batch, y: y_batch})
        acc_train = accuracy.eval(feed_dict={X: X_batch, y: y_batch})   # eval求accuracy的表达式的值
        acc_test = accuracy.eval(feed_dict={X: mnist.test.images.reshape((-1, n_steps, n_inputs)),
                                            y: mnist.test.labels})
        print(epoch+1, "Train accuracy:", acc_train, "Test accuracy:", acc_test)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值