用Tensorboard绘制自定义数据

问题描述 投票:-1回答:1

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

该指标只是一个标量,所以 我有一个数组的标量,我想用一个简单的图形来显示。 如。

enter image description here

我想把它实时显示在tensorboard上 就像我上面的例子。

先谢谢你

python machine-learning data-visualization tensorboard
1个回答
1
投票

如果你真的想使用 tensorboard,你可以开始关注一下 tensorflow网站 而这 数据营教程 在 tensorboard 上。

在 tensorflow 中,您可以使用 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()

tensorboard array of scalars

也就是说,如果你不打算在你的实现中使用tensorflow,我建议你直接使用matplotlib,因为这个库也能让你实时绘制数据。https:/youtu.beErcd-Ip5PfQ?t=444.

© www.soinside.com 2019 - 2024. All rights reserved.