使用 Tensorboard 绘制自定义数据

Question

我有一个 RL 算法的个人实现，它每 x 时间步生成一次性能指标。

该指标只是一个标量，所以我有一个标量数组，我想将其显示为一个简单的图形，例如：

我想像上面的例子一样在张量板上实时显示它。

提前致谢

Answer 1

如果您确实想使用张量板，您可以开始查看张量流网站和张量板上的这个datacamp 教程。

使用张量流，您可以用来summary.scalar绘制自定义数据（如示例），不需要特定的格式，因为摘要正在处理这一点，唯一的条件是data必须是a real numeric scalar value, convertible to a float32 Tensor.

import tensorflow as tf

import numpy as np

import os
import time

now = time.localtime()
subdir = time.strftime("%d-%b-%Y_%H.%M.%S", now)

summary_dir1 = os.path.join("stackoverflow", subdir, "t1")
summary_writer1 = tf.summary.create_file_writer(summary_dir1)

for cont in range(200):
    with summary_writer1.as_default():
        tf.summary.scalar(name="unify/sin_x", data=np.math.sin(cont) ,step=cont)
        tf.summary.scalar(name="unify/sin_x_2", data=np.math.sin(cont/2), step=cont)
    summary_writer1.flush()

也就是说，如果您不打算在实现中使用 Tensorflow，我建议您只使用 matplotlib，因为该库还使您能够实时绘制数据https://youtu.be/Ercd-Ip5PfQ?t=444。