使用 tensorflow 运行神经网络模型,我想绘制训练集和测试集的准确度得分。然而,图张量板向我显示看起来很奇怪:那里只有一个“准确性”选项卡,并且它在同一个数字上绘制了两个分数。所以基本上,x 轴上的每一步都有两个连接在一起的点。如何在该图上分别绘制两条线(训练精度和测试精度)?
这是我的代码片段:
loss_summary = tf.summary.scalar('损失', 损失)
acc_summary = tf.summary.scalar('准确度', 准确度)
summary_loss,summary_acc_train = sess.run([loss_summary, acc_summary], feed_dict={X: X_train, y: y_train})
Summary_acc_test = sess.run([acc_summary], feed_dict={X: X_test, y: y_test})
summary_writer.add_summary(summary_loss, 纪元)
summary_writer.add_summary(summary_acc_train, 纪元)
summary_writer.add_summary(summary_acc_test, epoch)
最佳答案
您需要创建两个不同的摘要编写器:
train_summary_writer = tf.summary.FileWriter(os.path.join(SUMMARIES_DIR, "train"), sess.graph)
validation_summary_writer = tf.summary.FileWriter(os.path.join(SUMMARIES_DIR, "validation"), sess.graph)
...
train_summary_writer.add_summary(summary_loss, epoch)
train_summary_writer.add_summary(summary_acc_train, epoch)
validation_summary_writer.add_summary(summary_acc_test, epoch)
关于tensorflow - 使用张量板绘制单独的学习曲线,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43996265/