把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()

 

結果入下圖所示

完結！