我正在构建一个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版本一起使用?