把tensorboard的多条训练损失曲线画在一张图上

在做实验的时候,通常需要对比分析不同模型的训练损失情况,tensorboard画出来的通常是不同的损失在不同的图上,那如果我们像把不同模型的训练损失画在同一张图进行对比分析,应该怎么处理呢?

可以通过matplotlib模块进行操作。

一、下载Tensorboard中的loss曲线的数据(下载方式如图所示)

二、用matplotlib模块画图

import csv
import matplotlib.pyplot as plt
import pandas as pd

df1 = pd.read_csv('/home/azhihong/JL-DCF-pytorch/log/run16-07/07-16-training loss.csv')  # csv文件所在路径
step1 = df1['Step'].values.tolist()
loss1 = df1['Value'].values.tolist()

df2 = pd.read_csv('/home/azhihong/JL-DCF-pytorch/log/run17-08/08-17-training loss.csv')
step2 = df2['Step'].values.tolist()
loss2 = df2['Value'].values.tolist()

plt.plot(step1, loss1, label='JL_DCF')
plt.plot(step2, loss2, label='Ours')
plt.legend(fontsize=16)  # 图注的大小
plt.show()

结果入下图所示

完结!

 

原文地址:https://www.cnblogs.com/elitphil/p/15158753.html