以下是使用Python和TensorFlow实现循环神经网络（RNN）的简单代码示例：
```python
import tensorflow as tf
# 定义超参数
learning_rate = 0.001
training_steps = 10000
batch_size = 128
display_step = 200
# 定义网络参数
num_input = 28
timesteps = 28
num_hidden = 128
num_classes = 10
# 定义输入张量
X = tf.placeholder("float", [None, timesteps, num_input])
Y = tf.placeholder("float", [None, num_classes])
# 定义权重和偏置变量
weights = {
'out': tf.Variable(tf.random_normal([num_hidden, num_classes]))
}
biases = {
'out': tf.Variable(tf.random_normal([num_classes]))
}
# 定义RNN模型
def RNN(x, weights, biases):
# 定义单层LSTM神经网络模型
lstm_cell = tf.contrib.rnn.BasicLSTMCell(num_hidden, forget_bias=1.0)
# 定义多层LSTM神经网络模型
# lstm_layers = [tf.contrib.rnn.BasicLSTMCell(num_hidden, forget_bias=1.0) for _ in range(num_layers)]
# lstm_cell = tf.contrib.rnn.MultiRNNCell(lstm_layers)
# 初始化状态
# state = lstm_cell.zero_state(batch_size, tf.float32)
# 将输入张量转为时间序列格式
x = tf.unstack(x, timesteps, 1)
# 定义循环神经网络模型
outputs, states = tf.contrib.rnn.static_rnn(lstm_cell, x, dtype=tf.float32)
# 输出层
output = tf.matmul(outputs[-1], weights['out']) + biases['out']
return output
# 构建模型
logits = RNN(X, weights, biases)
prediction = tf.nn.softmax(logits)
# 定义损失函数和优化器
loss_op = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=logits, labels=Y))
optimizer = tf.train.AdamOptimizer(learning_rate=learning_rate)
train_op = optimizer.minimize(loss_op)
# 定义评估模型的准确率
correct_pred = tf.equal(tf.argmax(prediction, 1), tf.argmax(Y, 1))
accuracy = tf.reduce_mean(tf.cast(correct_pred, tf.float32))
# 初始化变量
init = tf.global_variables_initializer()
# 训练模型
with tf.Session() as sess:
sess.run(init)
for step in range(1, training_steps+1):
batch_x, batch_y = mnist.train.next_batch(batch_size)
# 将数据转为RNN模型需要的格式
batch_x = batch_x.reshape((batch_size, timesteps, num_input))
# 运行优化器和损失函数
sess.run(train_op, feed_dict={X: batch_x, Y: batch_y})
if step % display_step == 0 or step == 1:
# 计算损失和准确率
loss, acc = sess.run([loss_op, accuracy], feed_dict={X: batch_x, Y: batch_y})
print("Step " + str(step) + ", Minibatch Loss= " + "{:.4f}".format(loss) + ", Training Accuracy= " + "{:.3f}".format(acc))
print("Optimization Finished!")
# 计算测试集准确率
test_data = mnist.test.images[:128].reshape((-1, timesteps, num_input))
test_label = mnist.test.labels[:128]
print("Testing Accuracy:", sess.run(accuracy, feed_dict={X: test_data, Y: test_label}))
```
这个示例使用MNIST数据集进行训练和测试，使用了单层LSTM神经网络模型，训练过程中使用Adam优化器和交叉熵损失函数。
最低0.47元/天 解锁文章
9077
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
