安装
pip install tensorboardX
还得安装TensorFlow
pip install tensorflow
使用
导包后往里面放数据就行,第一个参数是名称,第二个是y坐标,第三个是x坐标,通过add_scalar进行参数的添加。
import numpy as np
from tensorboardX import SummaryWriter
writer = SummaryWriter(log_dir='scalar')
for epoch in range(100):
writer.add_scalar('scalar/test', np.random.rand(), epoch)
writer.add_scalars('scalar/scalars_test', {'xsinx': epoch * np.sin(epoch), 'xcosx': epoch * np.cos(epoch)}, epoch)
writer.close()
结束后会生成scalar目录(log_dir所指),里面放了.event文件。
浏览生成的图片
tensorboard --logdir=scalar
- 输入上面的命令会开启后台服务器。
- 登录所给网址,可以进入tensorboard后台界面。
在这后台即可看到刚刚生成的两幅图。