Я новичок в программировании тензорного потока. Я хочу построить график точности обучения, потери обучения, точности проверки и потери проверки в следующей программе. Я использую tensorflow версии 1.x в google colab. Фрагмент кода выглядит следующим образом.
# hyperparameters
n_neurons = 128
learning_rate = 0.001
batch_size = 128
n_epochs = 5
# parameters
n_steps = 32
n_inputs = 32
n_outputs = 10
# build a rnn model
X = tf.placeholder(tf.float32, [None, n_steps, n_inputs])
y = tf.placeholder(tf.int32, [None])
cell = tf.nn.rnn_cell.BasicRNNCell(num_units=n_neurons)
output, state = tf.nn.dynamic_rnn(cell, X, dtype=tf.float32)
logits = tf.layers.dense(state, n_outputs)
cross_entropy = tf.nn.sparse_softmax_cross_entropy_with_logits(labels=y, logits=logits)
loss = tf.reduce_mean(cross_entropy)
optimizer = tf.train.AdamOptimizer(learning_rate=learning_rate).minimize(loss)
prediction = tf.nn.in_top_k(logits, y, 1)
accuracy = tf.reduce_mean(tf.cast(prediction, tf.float32))
# input data
x_test = x_test.reshape([-1, n_steps, n_inputs])
# initialize the variables
init = tf.global_variables_initializer()
# train the model
with tf.Session() as sess: sess.run(init)
n_batches = 100
for epoch in range(n_epochs):
for batch in range(n_batches):
sess.run(optimizer, feed_dict={X: x_train, y: y_train})
loss_train, acc_train = sess.run([loss, accuracy], feed_dict={X:
x_train, y: y_train})
print('Epoch: {}, Train Loss: {:.3f}, Train Acc:
{:.3f}'.format(epoch + 1, loss_train, acc_train))
loss_test, acc_test = sess.run([loss, accuracy], feed_dict={X:
x_test, y: y_test})
print('Test Loss: {:.3f}, Test Acc: {:.3f}'.format(loss_test,
acc_test))