from visdom import Visdom
line = viz.line(np.arange(10))
text = viz.text("<h1>convolution Nueral Network</h1>")
time_p, tr_acc, ts_acc, loss_p = [], [], [], []
time_p.append(epoch)
tr_acc.append(train_acc)
# ts_acc.append(acc)
loss_p.append(train_loss)
viz.line(X=np.column_stack((np.array(time_p), np.array(time_p))),
Y=np.column_stack((np.array(loss_p), np.array(tr_acc))),
win=line,
opts=dict(legend=["Loss", "TRAIN_acc"]))
# visdom text 支持html语句
viz.text("<p style='color:red'>epoch:{}</p><br><p style='color:blue'>Loss:{:.4f}</p><br>".format(epoch, train_loss),win=text)
运行前先打开环境终端 输入 python -m visdom.server 复制链接与Google
