python训练好的图片验证_用同一图表示TensorFlow中的训练和验证精度

您可以重用accurity节点,但需要使用两个不同的summarywriter,一个用于训练运行,另一个用于测试数据。此外,还必须为变量指定精度的标量摘要。accuracy_summary = tf.scalar_summary("Training Accuracy", accuracy)

tf.scalar_summary("SomethingElse", foo)

summary_op = tf.merge_all_summaries()

summaries_dir = '/me/mydir/'

train_writer = tf.train.SummaryWriter(summaries_dir + '/train', sess.graph)

test_writer = tf.train.SummaryWriter(summaries_dir + '/test')

然后在你的训练循环中,你进行正常的训练,并将你的总结记录在train_writer上。此外,每100次迭代在测试集上运行一次图形,并使用测试写入程序仅记录精度摘要。# Record train set summaries, and train

summary, _ = sess.run([summary_op, train_step], feed_dict=...)

train_writer.add_summary(summary, n)

if n % 100 == 0: # Record summaries and test-set accuracy

summary, acc = sess.run([accuracy_summary, accuracy], feed_dict=...)

test_writer.add_summary(summary, n)

print('Accuracy at step %s: %s' % (n, acc))

然后,您可以将TensorBoard指向父目录(summaries_dir),它将加载这两个数据集。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值