本节主要复习一下深度学习中这些常见的网络结构在tensorflow(1.x)中的使用,便于后续tensorflow的学习。
1. 全连接网络结构
全连接网络就是后层的每一个神经元均与前一层的神经元有关,按照上一节的推导,zl=w*al-1+b,然后再经过激活函数记得到了第l层的神经元al:
那么在tensorflow中的实现如下:
# w为权重,b为偏置,x为第l-1层的输出,这些均为tensor # 假设x输入[100],al-1输出[10],事先定义w和b变量 w = tf.Variable(tf.randon_normal([100, 10])) b = tf.Variable(tf.zeros([10])) # z = w*x + b z_l = tf.add(tf.matmal(x, w), b) # al = sigmoid(z_l) a_l = tf.nn.sigmoid(z_l)
这样就完成了1个全连接层,其中w,b是参数需要事先进行定义。
2. 卷积神经网络
卷积神经网络在【机器学习】中有一定的介绍,在那边主要使用的keras进行简单的实现,这里结合更多的例子,用tensorflow来进行实现。
首先定义几个输入图片的尺寸:
# shape依次为:batch_size, height, width, channel inputs = tf.Variable(tf.constant(1.0, shape=[1, 5, 5, 1])) input2 = tf.Variable(tf.constant(1.0, shape=[1, 5, 5, 2])) input3 = tf.Variable(tf.constant(1.0, shape=[1, 4, 4, 1]))
然后是filter,有以下几种filter:
""" filter的shape依次为:height, width, channel, output_channel height, width表示filter的大小, channel表示通道数,这个通道数与input的channel是一致的,也就是inputs的第四维, output_channel为输出的feature map个数,也就是filter的个数 """ # 第一个filter 大小2*2,只有1个filter filter1 = tf.Variable(tf.constant([-1.0, 0, 0, -1], shape=[2, 2, 1, 1])) # 第二个filter 大小2*2,有2个filter filter2 = tf.Variable(tf.constant([-1.0, 0, 0, -1, -1, 0, 0, -1], shape=[2, 2, 1, 2])) # 第三个filter 大小2*2,有3个filter filter3 = tf.Variable(tf.constant([-1.0, 0, 0, -1, -1, 0, 0, -1, -1, 0, 0, -1], shape=[2, 2, 1, 3])) # 第四个filter 大小2*2,有2个filter,输入的channel为2 filter4 = tf.Variable(tf.constant([-1.0, 0, 0, -1.0, -1.0, 0, 0, -1, -1.0, 0, 0, -1, -1.0, 0, 0, -1], shape=[2, 2, 2, 2])) # 第五个filter,大小2*2,有1个filter,输入的channel为1 filter5 = tf.Variable(tf.constant([-1.0, 0, 0, -1, -1.0, 0, 0, -1], shape=[2, 2, 2, 1]))
在tensorflow中卷积层的定义如下:
tf.nn.conv2d(input, filter, strides, padding, use_cuda_on_gpu)
input和filter说过了,strides是在卷积时在每一维上的步长,也就是1个四维的向量,一般在batch_size和output_channel上都是1。
padding是否填充边缘的元素,只能取“SAME”和“VALID”,“SAME”表示边缘补0,而“VALID”表示不补0,只有当stride为1时,“SAME”才能生成与输入一致的尺寸。
关于padding不同取值,决定着输出图片尺寸的大小,当为“VALID”情况时:
当padding为“SAME”时,输出大小与卷积核的大小没有关系,只与stride有关:
use_cuda_on_gpu表示是否在gpu上加速,默认为True。
那么接下来就根据上面的输入,根据不同的filter进行卷积:
# 输入5*5*1的图片,经过1个2*2的filter,步长为2, 那么输出为1个3*3的feature map op1 = tf.nn.conv2d(inputs, filter1, strides=[1, 2, 2, 1], padding="SAME") # output: shape=(1, 3, 3, 1) # 输入5*5*1的图片,经过2个2*2的filter,步长为2,那么输出为2个3*3的feature map op2 = tf.nn.conv2d(inputs, filter2, strides=[1, 2, 2, 1], padding="SAME") # output: shape=(1, 3, 3, 2) # 输入依旧5*5*1,经过3个2*2的filter,步长为2,输出则为3个3*3的feature map op3 = tf.nn.conv2d(inputs, filter3, strides=[1, 2, 2, 1], padding="SAME") # output: shape=(1, 3, 3, 3) # 输入为5*5*1, 但经过filter的channel为2,与输入的维度不一致,会报错 op_s = tf.nn.conv2d(inputs, filter4, strides=[1, 2, 2, 1], padding="SAME") # ValueError: Depth of input (1) is not a multiple of input depth of filter (2) for 'Conv2D_2' (op: 'Conv2D') with input shapes: [1,5,5,1], [2,2,2,2]. # 输入5*5*2的图片,有2个channel,经过2个2*2的filter,输出则为2个3*3的feature map op4 = tf.nn.conv2d(input2, filter4, strides=[1, 2, 2, 1], padding="SAME") # output: shape=(1, 3, 3, 2) # 输入5*5*2, 经过1个2*2的有2个channel的filter,输出则为1个3*3的feature map op5 = tf.nn.conv2d(input2, filter5, strides=[1, 2, 2, 1], padding="SAME") # output: shape=(1, 3, 3, 1) # 输入5*5*1,经过1个2*2的filter,padding=VALID,输出为(5-2+1)/2=2 v_op1 = tf.nn.conv2d(inputs, filter1, strides=[1, 2, 2, 1], padding="VALID") # output: shape=(1, 2, 2, 1) # 输入4*4*1,经过1个2*2的filter, padding=SAME,输出为4/2=2 op6 = tf.nn.conv2d(input3, filter1, strides=[1, 2, 2, 1], padding="SAME") # output: shape=(1, 2, 2, 1) # 输入4*4*1,经过1个2*2的filter, padding=VALID,输出为(4-2+1)/2=2 v_op6 = tf.nn.conv2d(input3, filter1, strides=[1, 2, 2, 1], padding="VALID") # output: shape=(1, 2, 2, 1)
卷积之后一般还有池化,下面是池化的方法:
tf.nn.max_pool(input, ksize, strides, padding, name)
tf.nn.avg_pool(input, ksize, strides, padding, name)
input是输入,ksize跟卷积中的filter一样,这里需要注意的是,第一维和第四维一般为1,因为一般不在batch和channel上做池化,输出channel一般与输出一致。其它与卷积中的参数一致。
下面举个例子,输入图片为:
img = tf.constant([[[0.0, 4.0], [0.0, 4.0], [0.0, 4.0], [0.0, 4.0]], [[1.0, 5.0], [1.0, 5.0], [1.0, 5.0], [1.0, 5.0]], [[2.0, 6.0], [2.0, 6.0], [2.0, 6.0], [2.0, 6.0]], [[3.0, 7.0], [3.0, 7.0], [3.0, 7.0], [3.0, 7.0]]]) img = tf.reshape(img, [1, 4, 4, 2])
# 输入均为4*4*2 # filter大小为2*2,步长为2,输出为(4-2+1)/2=2,输出channel依旧是2 pooling1 = tf.nn.max_pool(img, [1, 2, 2, 1], [1, 2, 2, 1], padding='VALID') # output: shape=(1, 2, 2, 2) # filter大小为2*2,步长为1,输出为(4-2+1)/1=3,输出channel依旧为2 pooling2 = tf.nn.max_pool(img, [1, 2, 2, 1], [1, 1, 1, 1], padding='VALID') # output: shape=(1, 3, 3, 2) # avg_pool跟max_pool类似,filter大小4*4,步长1,"SAME"与卷积核无关,输出为4/1=4 pooling3 = tf.nn.avg_pool(img, [1, 4, 4, 1], [1, 1, 1, 1], padding='SAME') # output: shape=(1, 4, 4, 2) # filter4*4,步长为4,"SAME"与卷积核无关,输出为4/4=1 pooling4 = tf.nn.avg_pool(img, [1, 4, 4, 1], [1, 4, 4, 1], padding='SAME') # output: shape=(1, 1, 1, 2)
此外还有反卷积操作和反池化操作,首先是反卷积,在【机器学习基础】中说过,其实反卷积就是一种卷积,不过是在边缘补0之后再进行卷积,在tensorflow中有函数来实现:
tf.nn.conv2d_transpose(input, filter, output_shape, strides, padding)
input表示输入;
filter是指在原图在卷积时所使用的filter;
output_shape反卷积后的尺寸大小,也就是原图经过卷积前的尺寸;
strides表示原图经过卷积时的strides;
padding表示原图在卷积时使用的padding
下面是示例:
# 原图片,并经过卷积 img = tf.Variable(tf.constant(1.0, shape=[1, 4, 4, 1])) filter = tf.Variable(tf.constant([1.0, 0, -1, -2], shape=[2, 2, 1, 1])) conv = tf.nn.conv2d(img, filter, strides=[1, 2, 2, 1], padding='VALID') cons = tf.nn.conv2d(img, filter, strides=[1, 2, 2, 1], padding='SAME') # 反卷积过程,所有参数与卷积过程保持一致,输出则为img的尺寸 contv = tf.nn.conv2d_transpose(conv, filter, [1, 4, 4, 1], strides=[1, 2, 2, 1], padding="VALID") conts = tf.nn.conv2d_transpose(cons, filter, [1, 4, 4, 1], strides=[1, 2, 2, 1], padding="SAME")
然后接下来是反池化,在【机器学习】基础中,反池化的原理也已经过了,这里就不再赘述。
这里直说实现,对于反池化的操作,有一点比较麻烦的在池化过程中,我们要知道哪一个元素被选为最大,在反池化时,要将对应的元素还原,其他元素置为0。因此,池化时还需要记录被选中的那一个元素。
因此,这里需要对池化操作重新写,tensorflow中提供了记录最大值的方法tf.nn.max_pool_with_argmax方法,下面是修改后的池化操作,放在这里后面可以直接用:
# 返回值是池化后的网络以及对应的索引值 def max_pool_with_argmax(net, stride): _, mask = tf.nn.max_pool_with_argmax(net, ksize=[1, stride, stride, 1], strides=[1, stride, stride, 1], padding='SAME') mask = tf.stop_gradient(mask) net = tf.nn.max_pool(net, ksize=[1, stride, stride, 1], strides=[1, stride, stride, 1], padding='SAME') return net, mask
有了返回的索引,就可以进行反池化操作了,反池化没有现成的函数,需要自己实现,下面是实现过程,记下便于后续使用:
def unpool(net, mask, stride): ksize = [1, stride, stride, 1] input_shape = net.get_shape().as_list() # 计算new_shape output_shape = (input_shape[0], input_shape[1] * ksize[1], input_shape[2]*ksize[2], input_shape[3]) one_like_mask = tf.ones_like(mask) batch_range = tf.reshape(tf.range(output_shape[0], dtype=tf.int64), shape=[input_shape[0], 1, 1, 1]) b = one_like_mask * batch_range y = mask // (output_shape[2] * output_shape[3]) x = mask % (output_shape[2] * output_shape[3]) // output_shape[3] feature_range = tf.range(output_shape[3], dtype=tf.int64) f = one_like_mask * feature_range update_size = tf.size(net) indices = tf.transpose(tf.reshape(tf.stack([b, y, x, f]), [4, update_size])) values = tf.reshape(net, [update_size]) ret = tf.scatter_nd(indices, values, output_shape) return ret
下面举个例子来进行池化和反池化操作:
img = tf.constant([[[0.0, 4.0], [0.0, 4.0], [0.0, 4.0], [0.0, 4.0]], [[1.0, 5.0], [1.0, 5.0], [1.0, 5.0], [1.0, 5.0]], [[2.0, 6.0], [2.0, 6.0], [2.0, 6.0], [2.0, 6.0]], [[3.0, 7.0], [3.0, 7.0], [3.0, 7.0], [3.0, 7.0]]]) img = tf.reshape(img, [1, 4, 4, 2]) # 池化操作,stride=2,一般在max_pool时设定步长与stride一致 pool, mask = max_pool_with_argmax(img, 2) # 反池化 img2 = unpool(pool, mask, 2) with tf.Session() as sess: print('img', img) result = sess.run(img2) print('img2', result)
可以看到img2和img在尺寸上是一致的,但是数值不太一样,这在前面说过,对与不是最大值的部分填充0。
有了上面的基本结构,就可以构建一个简单的CNN深度网络了,下面是是一个基本的网络结构示例:
# 先定义初始化参数的函数,不必每次遇到参数定义就初始化一次,直接调用 def weight_variable(shape): initial = tf.Variable(tf.truncated_normal(shape, std=0.1)) return initial # 同样定义bias def bias_variable(shape): initial = tf.Variable(tf.constant(0.1, shape=shape)) return initial # 定义卷积函数,每次直接调用即可 def conv2d(x, w): return tf.nn.conv2d(x, w, strides=[1, 1, 1, 1], padding="SAME") # 输入占位符,图片大小24*24*3 x = tf.placeholder(tf.float32, [batch_size, 24, 24, 3]) # 输出占位符,10个类别 y = tf.placeholder(tf.float32, [batch_size, 10]) # 第一个卷积层的卷积核参数,有64个5*5*3的filter w_conv1 = weight_variable([5, 5, 3, 64]) # 卷积后需要加上一个bias, 卷积后为64个feature map,所以bias维度为64 b_conv1 = bias_variable([64]) # 卷积操作,一般卷积层后会接RELU,输出的尺寸为24/1=24,channel为64 h_conv1 = tf.relu(conv2d(x, w_conv1) + b_conv1) # 然后是池化操作,池化后尺寸为12*12*64 h_pool1, mask1 = max_pool_with_argmax(h_conv1, 2) # 第二层卷积的卷积核参数,有64个5*5*64的 w_conv2 = weight_varibale([5, 5, 64, 64]) b_conv2 = bias_varibale([64]) # 第二次卷积操作,卷积后依旧12*12*64 h_conv2 = tf.relu(conv2d(h_pool1, w_conv2) + b_conv2) # 第二次池化,池化后尺寸6*6*64 h_pool2, mask2 = max_pool_with_argmax(h_conv2, 2) # 接下来反池化操作,对h_pool2进行反池化,得到第二次卷积后的结果 t_conv2 = unpool(h_pool2, mask2, 2) # 然后是反卷积,反卷积后,与第一次池化结果对应 # 反卷积操作中要减掉bias t_pool1 = tf.nn.conv2d_transpose(t_conv2 - b_conv2, w_conv2, h_pool1.shape, [1, 1, 1, 1]) # 再次反池化,返回到第一次卷积后的结果 t_conv1 = unpool(t_pool1, mask1, 2) # 再次反卷积,反卷积后即回到初始的x t_x = tf.nn.conv2d_transpose(t_conv1 - b_conv1, w_conv1, x.shape, [1, 1, 1, 1]) # 紧接着在第二次池化后,在进行一次卷积池化操作 w_conv3 = weight_varibale([5, 5, 64, 10]) b_conv3 = bias_variable([10]) # 第三次卷积后的输出为6*6,channel为10 h_conv3 = tf.nn.relu(conv2d(h_pool2, w) + b_conv3) # 最后经过一个6*6的avg_pool,输出为1*1,channel为10 h_pool3 = tf.nn.avg_pool(h_conv3, [1, 6, 6, 1], strides=[1, 6, 6, 1], padding="SAME") # 将h_pool3展开,输入到softmax h_pool3_flat = tf.reshape(h_pool3, [-1, 10]) # 可以再接几层fc fc1 = tf.layers.dense(h_pool3_flat, 64, activation=tf.nn.tanh) fc2 = tf.layers.dense(fc1, 64, activation=tf.nn.tanh) y_pred = tf.layers.dense(fc2, 10, activation=tf.nn.softmax)
这样,一个三层的卷积神经网络就好了,在卷积后的三层其实可以再接几层fc层。
以上就是一个较为完整的网络的实现了。
这里需要注意:反卷积和反池化的作用主要是为了实现可视化、或者还原原图像分辨率的问题,并没有参与训练。
另外,关于卷积后需要加bias的问题,当卷积后接Bath Normalization(后面会说)时,是不需要加bias的。
3. RNN
RNN中有多种结构,除了基本的basic RNN,还有LSTM、GRU。在使用RNN时,需要预先定义好cell,然后再将cell连接起来,构成RNN网络。
tensorflow中包含了上面几种cell的定义:
3.1basic RNN
首先是basic RNN,该cell的定义如下:
tf.contrib.rnn.BasicRNNCell(num_units, input_size=None, activation=tanh)
num_units表示隐藏节点的个数,也就是ht的维度,input_size废弃了不再使用。
定义一个basic RNN cell也比较简单,只需指定好n_hidden的大小即可:
basic_rnn_cell = tf.contrib.rnn.BasicRNNCell(n_hidden)
3.2 LSTM
然后是LSTM,也是类似,LSTM定义为:
tf.contrib.rnn.BasicLSTMCell(num_units, forget_bias=1.0, state_in_tuple, activation)
num_units一样表示隐藏层节点的个数,forget_bias是添加到遗忘门的偏置,表示保留多少信息,默认为1。
state_in_tuple是否将记忆c和h以tuple的形式分开,当为True时输出tuple(c=array(), h=array()),当为False时两个连接起来[batch, 2n],该参数即将废弃。
定义一个basic LSTMcell:
basic_lstm = tf.contrib.rnn.BasicLSTMCell(n_hidden, forget_bias=1.0)
这样就定义好了一个基本的LSTM结构。
还有另一个LSTM的高级版本版本,除了上面的参数外,还有一些其他的参数:
3.3 GRU
GRU的定义也是一样的:
tf.contrib.rnn.GRUCell(num_units, input_size=None, activation=tanh)
basic_gru = tf.contrib.rnn.GRUCell(n_hidden)
3.4 MultiRNN
上面就是几个基本cell的定义,当然,cell不单单是某一种,可以是多种组合起来的多层cell,其定义:
tf.contrib.rnn.MultiRNNCell(cells, state_in_tuple)
cells就是一个cell的列表,该列表为一系列basic cell组成,cells=[cell1, cell2]就表示有两层数据经过cell1后还要经过cell2.
state_in_tuple于basic rnn中类似,将记忆cell与隐藏状态输出h组成一个tuple。
注意:在使用MultiCell时,cells不能直接用[cell]×n的写法,如果不使用作用域会报错,而是要用append的写法。
创建一个多层的cell:
cell1 = tf.contrib.rnn.BasicLSTMCell(n_hidden) cell2 = tf.contrib.rnn.GRUCell(n_hidden) #如果两个cell的n_hidden,则输出以最后一个节点为准 multi_rnn = tf.contrib.rnn.MultiRNNCell([cell1, cell2])
3.5 RNN的连接
上面构建出基本的RNN cell时只是某一个时刻的状态,接下来还要把Cell连接起来,才能构成完整的RNN网络。
RNN的连接方式可以分为静态、动态,单向RNN和双向RNN,再结合上面的单层的cell和多层的cell,那么RNN各种条件自由组合,可以有8中不同的连接。
这里所谓的静态和动态,是指在创建RNN时,RNN预先创建好的结构,这时需要输入的长度要与该结构保持一致;
因此静态的因为预先创建好,会花费较多的时间和占用空间,但其优点是可以查看任一时刻的输出,便于调试;
而动态则是指比较灵活地根据输入去创建cell,通过循环来创建的。
因此动态的创建快,节省内存,便于不同长度的输入的批处理,但缺点是只能查看序列最终状态,不方便调试。
静态的创建方式:
tf.contrib.rnn.static_rnn(cells, input, initial_state=None, sequence_length=None, scope=None)
cells为前面说的cell;
input为输入;
initial_state初始化状态参数,一般初始化为0时,不需刻意指定,要是想要初始化指定的值用initial_state=LSTMStateTuple(c_state, h_state)
sequence_lenght一般不需要指定,会根据输入自动识别;
scope:命名空间。
动态的创建方式:
tf.contrib.rnn.dynamic_rnn(cells, input, initial_state=None, sequence_length=None, time_major, scope=None)
参数基本与静态的一致,这里多了一个time_major,默认为False时,input_shape=[batch_size, max_time, …..], 当为True时,input_shape=[max_time, batch_size, ….]
接下来就是各种RNN的实现方式:
3.5.1 单向的RNN:
① 单层静态单向RNN
# 输入x为n_step*n_input,即n_step个时间点,每个时间点的维度为n_input x = tf.placeholder(tf.float32, [None, n_steps, n_input]) # 先创建一个lstm cell lstm_cell = tf.contrib.rnn.BasicLSTMCell(n_hidden) # 静态单层RNN # 需要先转换x的形状为预设形状 x1 = tf.unstack(x, n_steps, 1) outputs, states = tf.contrib.rnn.static_rnn(lstm_cell, x1, dtype=tf.float32) # outputs包括结果和cell的状态,只需关注结果,最后一维才是结果 pred = tf.contrib.layers.fully_connected(couputs[-1], 10, activation_fn=None)
②多层静态单向RNN
x = tf.placeholder(tf.float32, [None, n_steps, n_input]) # 先创建多层的cells cells = [] for i in range(3): cells.append(tf.contrib.rnn.BasicLSTMCell(n_hidden)) multi_cell = tf.contrib.rnn.MultiRNN(cells) x1 = tf.unstack(x, n_steps, 1) # 多层静态单向RNN outputs, states = tf.contrib.rnn.static_rnn(multi_cell, x1, dtype=tf.float32) pred = tf.contrib.layers.fully_connected(outputs[-1], 10, activation_fn=None)
③单层动态单向RNN
x = tf.placeholder(tf.float32, [None, n_steps, n_input]) # 先创建basic cell lstm_cell = tf.contrib.rnn.BasicLSTMCell(n_hidden) # 动态单层单向RNN outputs, state = tf.contrib.rnn.dynamic_rnn(lstm_cell, x, dtype=tf.float32) outputs = tf.transpose(outputs, [1, 0, 2]) pred = tf.contrib.layers.fully_connected(outputs[-1], 10, activation_fn=None)
④多层动态单向RNN
x = tf.placeholder(tf.float32, [None, n_steps, n_input]) # basic cell lstm_cell = tf.contrib.rnn.BasicLSTMCell(n_hidden*2) gru_cell = tf.contrib.rnn.GRUCell(n_hidden) # 构建多层cell multi_cells = tf.contrib.rnn.MultiRNNCell([lstm_cell, gru_cell]) # 多层动态单向RNN outputs, state = tf.contrib.rnn.dynamic_rnn(multi_cells, x, dtype=tf.float32) outputs = tf.transpose(outputs, [1, 0, 2]) pred = tf.contrib.layers.fully_connected(outputs[-1], 10, activation_fn=None)
从上面的代码可以看到,静态和动态的区别再输入和输出的格式转化,主要有两点:
1、静态的输入需要将输入转换成list的形式,而动态网络不需要转换。这是因为静态是预先设定好的,必须按照设定的输入;
2、动态的输出需要进行转置操作,而静态的则不需要。这是因为动态网络输出形式为[batch_size, max_time, …],
转置[1, 0, 2]是将第0维max_time放在后面,将batch_size放在第1维,这样转换后变为[max_time, batch_size,…..], 取最后一维时就是最后一个时间点的数据[batch_size, ……]
3.5.2 双向的RNN
双向的RNN在单层和多层之间的定义还是有所差别的,对于单层的CELL的静态RNN定义:
tf.contrib.rnn.static_bidirectional_rnn(cell_fw, cell_bw, inputs)
对于单层的动态RNN定义如下:
tf.nn.bidirectional_dynamic_rnn(cell_fw, cell_bw, inputs)
而对于多层的静态RNN定义:
tf.contrib.rnn.stack_bidirectional_rnn(cells_fw, cells_bw, inputs)
对于多层动态的RNN定义:
tf.contrib.rnn.stack_bidirectional_dynamic_rnn(cells_fw, cells_bw, inputs)
从上面的四个定义可以看出:
无论是单层还是多层,均需一个前向的cell和一个反向的cell,且前向的cell和反向的cell的结构必须保持一致;
对于单层的定义没有“stack”,多层的需要带“stack”,其中单层动态的来自于tf.nn模块;
而对于多层的cells可以接受列表的形式,也可以接受Multi形式的cell,后面举例说明。
① 单层静态双向RNN
x = tf.placeholder(tf.float32, [None, n_steps, n_input]) # 对于静态的而言,输入要调整为list x1 = tf.unstack(x, n_stpes, 1) # 先创建basic cell,前向和反向 cell_fw = tf.contrib.rnn.BasicLSTMCell(n_hidden) cell_bw = tf.contrib.rnn.GRUCell(n_hidden) # 单层静态双向RNN outputs, _, _ = tf.contrib.rnn.static_bidirectional_rnn(cell_fw, cell_bw, x1) pred = tf.contrib.layers.fully_connected(outputs[-1], 10, activation_fn=None)
②单层动态双向RNN
x = tf.placeholder(tf.float32, [None, n_steps, n_input]) cell_fw = tf.contrib.rnn.BasicLSTMCell(n_hidden) cell_bw = tf.contrib.rnn.GRUCell(n_hidden) outputs, _, _ = tf.nn.bidirectional_dynamic_rnn(cell_fw, cell_bw, x, dtype=tf.float32) # 动态rnn的输出需要进行转化为[max_time, batch_size, ....] outputs = tf.transpose(outputs, [1, 0, 2]) pred = tf.contrin.layers.fully_connected(outputs[-1], 10, activation_fn=None)
③多层静态双向RNN
x = tf.placeholder(tf.float32, [None, n_steps, n_input]) x1 = tf.unstack(x, n_steps, 1) cell_fw = tf.contrib.rnn.BasicLSTMCell(n_hidden) cell_bw = tf.contrib.rnn.GRUCell(n_hidden) # 创建多层静态双向RNN,多层的只需将cell放入列表即可,但必须保证前向与后向结构一致 outputs, _, _ = tf.contrib.rnn.stack_bidirectional_rnn([cell_fw], [cell_bw], x1, dtype=tf.float32) pred = tf.contrib.layers.fully_connected(outputs[-1], 10, activation_fn=None) # 多层的也接受multi cell stack_cells_fw = [] stack_cells_bw = [] for i in range(3): stack_cells_fw.append(tf.contrib.rnn.BasicLSTMCell(n_hidden)) stack_cells_bw.append(tf.contrib.rnn.BasicLSTMCell(n_hidden)) multi_cells_fw = tf.contrib.rnn.MultiRNNCell(stack_cells_fw) multi_cells_bw = tf.contrib.rnn.MultiRNNCell(stack_cells_bw) outputs, _, _ = tf.contrib.rnn.stack_bidirectional_rnn([multi_cells_fw], [multi_cells_bw], x1, dtype=tf.float32) ......
④多层动态双向RNN
x = tf.placeholder(tf.float32, [None, n_steps, n_input]) stack_cells_fw = [] stack_cells_bw = [] for i in range(3): stack_cells_fw.append(tf.contrib.rnn.BasicLSTMCell(n_hidden)) stack_cells_bw.append(tf.contrib.rnn.BasicLSTMCell(n_hidden)) # 多层动态双向RNN outputs, _, _ = tf.contrib.rnn.stack_bidirectional_dynamic_rnn(stack_cells_fw, stack_cells_bw, x, dtype=tf.float32) outputs = tf.transpose(outputs, [1, 0, 2]) pred = tf.contrib.layers.fully_connected(outputs[-1], 10, activation_fn=None) # 也接受Multi cell multi_cells_fw = tf.contrib.rnn.MultiRNNCell(stack_cells_fw) multi_cells_bw = tf.contrib.rnn.MultiRNNCell(stack_cells_bw) outputs, _, _ = tf.contrib.rnn.stack_bidirectional_dynamic_rnn([multi_cells_fw], [multi_cells_bw], x, dtype=tf.float32) ......
以上就是RNN的基本结构的实现过程,下面利用MNIST数据集,使用RNN训练一个模型:
import tensorflow as tf from tensorflow.contrib import rnn from tensorflow.examples.tutorials.mnist import input_data # 读取数据 mnist = input_data.read_data_sets("/data/", ont_hot=True) # 参数设置 learning_rate = 0.001 traning_iters = 100000 batch_size = 128 display_step = 10 # 网络的参数 # 图片尺寸28*28,当做序列长度为28,每一个时刻的维度为28 n_inputs = 28 # 每一个序列的维度 n_steps = 28 # 序列长度 n_hidden = 128 # 隐藏层的输出维度 n_classes = 10 # 0~9 10个类别 # 初始化图 tf.reset_default_graph() # 输入、输出的占位符 x = tf.placeholder("float", [None, n_steps, n_input]) y = tf.placeholder("float", [None, n_classes]) # 这里采用多层双向动态RNN # 定义前向、后向basic cells stack_cells_fw = [] stack_cells_bw = [] for i in range(3): stack_cells_fw.append(rnn.BasicLSTMCell(n_hidden)) stack_cells_bw.append(rnn.GURCell(n_hidden)) # 转为multi_cells multi_cells_fw = rnn.MultiRNNCell(stack_cells_fw) multi_cells_bw = rnn.MultiRNNCell(stack_cells_bw) # 创建多层双向动态RNN outputs, _, _ = rnn.stack_bidirectional_dynamic_rnn([multi_cells_fw], [multi_cells_bw], x, dtype=tf.float32) outputs = tf.transpose(outputs, [1, 0, 2]) # 输出值后接一个全连接网络,输出为10维,classes pred = tf.contrib.layers.fully_connected(outputs[-1], n_classes, activation_fn=None) # 计算loss,采用cross_entropy loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=pred, labels=y)) # 梯度下降 optimizer = tf.train.AdamOptimizer(learning_rate=learning_rate).minimize(loss) # 估计正确率 correct_pred = tf.equal(tf.argmax(pred, 1), tf.argmax(y, 1)) accuracy = tf.reduce_mean(tf.cast(corrected_pred, tf.float32)) # 启动session开始训练 with tf.Session() as sess: # 初始化 sess.run(tf.global_variables_initializer()) step = 1 while step * batch_size < training_iters: # 批数据 batch_x, batch_y = mnist.train.next_batch(batch_size) # 将batch_x 转成x的形状 batch_x = batch_x.reshape([batch_size, n_steps, n_input]) sess.run(optimizer, feed_dict={x:batch_x, y: batch_y}) # 打印训练过程 if step%display_step == 0: # 计算这一轮的精度 acc = sess.run(accuracy, feed_dict={x: batch_x, y: batch_y}) # 计算损失 cost = sess.run(loss, feed_dict={x: batch_x, y: batch_y}) print('Iter' + str(step * batch_size) + ', Mini-batch loss= ', + '{:.6f}'.format(cost) + ', Training Accuracy=' + '{:.5f}'.format(acc)) step += 1 print('Finished') # 计算在测试集上的准确率 test_len = 128 test_data = mnist.test.images[:test_len].reshape([-1, n_steps, n_input]) test_labels = mnist.test.labels[:test_len] print('Testing Accuracy:', sess.run(acc, feed_dict={x: test_data, y: test_labels}))