由于TF版本导致建立LSTM模型时出错

时间:2019-01-03 13:01:00

标签: python tensorflow lstm recurrent-neural-network

我正在构建一个LSTM模型来预测时间序列的演变,但是由于我在自己的资源中引用了某些先前版本的Tensorflow,所以我有一个错误。 我的代码是:

from tensorflow.contrib import rnn



def LSTM(x):
    layer = {'weights':tf.Variable(tf.random_normal([rnn_size, 1])),'biases':tf.Variable(tf.random_normal([1]))}
    x = tf.transpose(x, [1,0,2])
    x = tf.reshape(x, [-1, chunk_size])
    x = tf.split(x, n_chunks, 0)
    lstm_cell = tf.nn.rnn_cell.LSTMCell(rnn_size, state_is_tuple=True, activation=tf.nn.relu)
    outputs, states = rnn.rnn(lstm_cell, x, dtype=tf.float32)
    output = tf.matmul(outputs[-1], layer['weights']) + layer['biases']
    return output

最后我得到错误消息: 模块'tensorflow.contrib.rnn'没有属性'rnn'

如何修改我的代码以使其与tensorflow 1.12.0版本一起使用?

0 个答案:

没有答案