一、概念
循环神经网络(Recurrent Neural Network,RNN):
它能结合数据点之间的特定顺序和幅值大小等多个特征,来处理序列数据。更重要的是输入序列可以是任意长度的。
RNNs的目的是用来处理序列数据。在传统的神经网络模型中,是从输入层到隐含层再到输出层,层与层之间是全连接的,每层之间的节点是无连接的。但是这种普通的神经网络对于很多问题却无能为力。例如,你要预测句子的下一个单词是什么,一般需要用到前面的单词,因为一个句子中前后单词并不是独立的。RNN之所以称为循环神经网络,即一个序列当前的输出与前面的输出也有关。具体的表现形式为网络会对前面的信息进行记忆并应用于当前输出的计算中,即隐藏层之间的节点不再无连接而是有连接的,并且隐藏层的输入不仅包括输入层的输出还包括上一时刻隐藏层的输出。下图便是一个典型的RNN:
二、代码
- from tensorflow.examples.tutorials.mnist import input_data
- mnist = input_data.read_data_sets("MNIST_data/",one_hot=True)
- import tensorflow as tf
- import numpy as np
- from tensorflow.contrib import rnn
- '''''
- 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 # 隐含层的特征数
- n_classes = 10 # 输出的数量,因为是分类问题,0~9个数字,这里一共有10个
- # 构建tensorflow的输入X的placeholder
- x = tf.placeholder("float", [None, n_steps, n_input])
- # tensorflow里的LSTM需要两倍于n_hidden的长度的状态,一个state和一个cell
- # Tensorflow LSTM cell requires 2x n_hidden length (state & cell)
- istate = tf.placeholder("float", [None, 2 * n_hidden])
- # 输出Y
- y = tf.placeholder("float", [None, n_classes])
- # 随机初始化每一层的权值和偏置
- weights = {
- 'hidden': tf.Variable(tf.random_normal([n_input, n_hidden])), # Hidden layer weights
- 'out': tf.Variable(tf.random_normal([n_hidden, n_classes]))
- }
- biases = {
- 'hidden': tf.Variable(tf.random_normal([n_hidden])),
- 'out': tf.Variable(tf.random_normal([n_classes]))
- }
- #构建RNN
- def RNN(_X, _istate, _weights, _biases):
- # 规整输入的数据
- _X = tf.transpose(_X, [1, 0, 2]) # permute n_steps and batch_size
- _X = tf.reshape(_X, [-1, n_input]) # (n_steps*batch_size, n_input)
- # 输入层到隐含层,第一次是直接运算
- _X = tf.matmul(_X, _weights['hidden']) + _biases['hidden']
- # 之后使用LSTM
- #lstm_cell = rnn_cell.LayerNormBasicLSTMCell(n_hidden, forget_bias=1.0)
- lstm_cell = rnn.BasicLSTMCell(n_hidden,state_is_tuple=False)
- # 28长度的sequence,所以是需要分解位28次
- _X = tf.split(_X, n_steps, 0) # n_steps * (batch_size, n_hidden)
- #x = tf.split(x, n_steps, 0) # tf.split(value, num_or_size_splits, axis) versions > 0.12.0
- # 开始跑RNN那部分
- outputs, states = rnn.static_rnn(lstm_cell, _X, initial_state=_istate)
- #rnn.rnn.dynamic_rnn()
- return tf.matmul(outputs[-1],_weights['out']) + biases['out']
- pred = RNN(x, istate, weights, biases)
- # 定义损失和优化方法,其中算是为softmax交叉熵,优化方法为Adam
- cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=pred, labels=y)) # Softmax loss
- optimizer = tf.train.AdamOptimizer(learning_rate=learning_rate).minimize(cost) # Adam Optimizer
- # 进行模型的评估,argmax是取出取值最大的那一个的标签作为输出
- correct_pred = tf.equal(tf.argmax(pred, 1), tf.argmax(y, 1))
- accuracy = tf.reduce_mean(tf.cast(correct_pred, tf.float32))
- # 初始化
- init = tf.initialize_all_variables()
- # 开始运行
- with tf.Session() as sess:
- sess.run(init)
- step = 1
- # 持续迭代
- while step * batch_size < training_iters:
- # 随机抽出这一次迭代训练时用的数据
- batch_xs, batch_ys = mnist.train.next_batch(batch_size)
- # 对数据进行处理,使得其符合输入
- batch_xs = batch_xs.reshape((batch_size, n_steps, n_input))
- # 迭代
- sess.run(optimizer, feed_dict={x: batch_xs, y: batch_ys,
- istate: np.zeros((batch_size, 2 * n_hidden))})
- # 在特定的迭代回合进行数据的输出
- if step % display_step == 0:
- # Calculate batch accuracy
- acc = sess.run(accuracy, feed_dict={x: batch_xs, y: batch_ys,
- istate: np.zeros((batch_size, 2 * n_hidden))})
- # Calculate batch loss
- loss = sess.run(cost, feed_dict={x: batch_xs, y: batch_ys,
- istate: np.zeros((batch_size, 2 * n_hidden))})
- print ("Iter " + str(step * batch_size) + ", Minibatch Loss= " + "{:.6f}".format(loss) + \
- ", Training Accuracy= " + "{:.5f}".format(acc) )
- step += 1
- print ("Optimization Finished!")
- # 载入测试集进行测试
- test_len = 256
- test_data = mnist.test.images[:test_len].reshape((-1, n_steps, n_input))
- test_label = mnist.test.labels[:test_len]
- print ("Testing Accuracy:", sess.run(accuracy, feed_dict={x: test_data, y: test_label,
- istate: np.zeros((test_len, 2 * n_hidden))}) )
三、代码解释
class tf.contrib.rnn.BasicLSTMCell
BasicLSTMCell是最简单的一个LSTM类, lstm=rnn.BasicLSTMCell(lstm_size,forget_bias,state_is_tuple)
Args:
lstm_size :LSTM Cell 个数
说明:lstm_size是指一个Cell中神经元的个数,并不是循环层的Cell个数。循环层的Cell数目由如下代码中的time_step_size确定(X_split中划分出的arrays数量为循环层的CEll个数):
X_split=tf.split(XR,time_step_size,0)
以相邻的时间步分割批数据,RNN会同时从不同位置开始训练时间序列:在示例中分别从4到6、从16到18和从28到30.
在时间序列数据中,在三个位置同时开启训练,所以在前向传播时需要保存三个状态。