把Python变量记录到TensorFlow的summary中

在用tensorboard可视化时,遇到过一个坑:打算每训练一定步数,将验证集准确率记录下来。

一般做法如下:

...
# 1、计算准确率
correct_prediction = tf.equal(tf.argmax(logits, 1), tf.argmax(y, 1))
accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))

# 2、添加scalar,汇总
tf.summary.scalar('accuracy', accuracy)
merged = tf.summary.merge_all()

...
# 3、运行,写入log
summary, acc = sess.run([merged, accuracy], feed_dict=feed_dict(False))  
test_writer.add_summary(summary, i)  

上面的做法有个问题,在计算准确率时传入的是整个数据集,如果数据集很大,内存会不够用。一般需要分多个batch,每次输入一个batch,计算预测正确的样本数,然后求和除以总样本数:

steps_per_epoch = data.num_examples // FLAGS.batch_size
num_examples = steps_per_epoch * FLAGS.batch_size
true_count = 0
for i in range(steps_per_epoch):
    images_feed, labels_feed = data.next_batch(FLAGS.batch_size)
    true_count += sess.run(correct, feed_dict={images_pl: images_feed, labels_pl:labels_feed, training_pl: False})
validate_accuracy = true_count / num_examples

但是这样得到的结果validate_accuracy是一个变量,如何把普通的Python变量加到summary中?

官方APIhttps://tensorflow.google.cn/api_docs/python/tf/summary/FileWriter#add_summary中描述如下:

You can pass the result of evaluating any summary op, using tf.Session.run or tf.Tensor.eval, to this function. Alternatively, you can pass a tf.Summary protocol buffer that you populate with your own data. The latter is commonly done to report evaluation results in event files.

实现如下:

summary = tf.Summary()
summary.value.add(tag='val_acc', simple_value=validate_accuracy)
summary_writer.add_summary(summary, i)

最后运行结果:

  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
您可以使用 TensorFlow 的 `tf.summary` 模块来记录和可视化多个变量在同一个图的值。以下是一个示例: ```python import tensorflow as tf # 创建两个变量 var1 = tf.Variable(1.0, name="var1") var2 = tf.Variable(2.0, name="var2") # 创建一个操作,将变量相加 sum_vars = tf.add(var1, var2, name="sum_vars") # 创建一个 FileWriter 对象来将日志写入磁盘 writer = tf.summary.FileWriter('./logs') # 创建一个合并所有变量摘要的操作 merged_summary = tf.summary.merge_all() # 在会话运行变量初始化操作 init = tf.global_variables_initializer() with tf.Session() as sess: sess.run(init) # 运行 merged_summary 操作并将结果写入日志 summary = sess.run(merged_summary) writer.add_summary(summary) # 运行 sum_vars 操作并将结果打印出来 result = sess.run(sum_vars) print(result) ``` 在这个示例,我们创建了两个变量 `var1` 和 `var2`,然后创建了一个操作 `sum_vars`,将这两个变量相加。接下来,我们创建了一个 FileWriter 对象来将日志写入磁盘,然后创建了一个合并所有变量摘要的操作 `merged_summary`。在会话运行变量初始化操作,然后运行 merged_summary 操作,并将结果写入日志。最后,我们运行 sum_vars 操作并将结果打印出来。 注意,这个示例只有一个操作(sum_vars),但是您可以将多个操作添加到摘要,并在会话运行 merged_summary 操作来记录和可视化所有变量的值。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值