通過python畫出loss曲線

• 將訓(xùn)練過程中的running loss保存到文件
• 1.讀取txt文件
• 2.取出相應(yīng)列的數(shù)據(jù)作為繪圖的x和y
• 3.先創(chuàng)建一幅圖，再在這幅圖上添加一個(gè)小圖
• 4.畫出整體的loss曲線

將訓(xùn)練過程中的running loss保存到文件

for x in range(100):with open('loss_records.txt', 'a') as f:f.write('%d %.3f\n' % (x, x + 1))with open('accurate_records.txt', 'a') as f:f.write('%d %.3f\n' % (x, x + 10))

注意：
模式：a
打開一個(gè)文件用于追加。如果該文件已存在，文件指針將會(huì)放在文件的結(jié)尾。也就是說，新的內(nèi)容將會(huì)被寫入到已有內(nèi)容之后。如果該文件不存在，創(chuàng)建新文件進(jìn)行寫入。

1.讀取txt文件

現(xiàn)在有兩個(gè)模型訓(xùn)練結(jié)果的records.txt文件

import matplotlib.pyplot as plt import numpy as npdata1_loss = np.loadtxt("accurate_records.txt") data2_loss = np.loadtxt("loss_records.txt")

2.取出相應(yīng)列的數(shù)據(jù)作為繪圖的x和y

數(shù)據(jù)里第一列是訓(xùn)練步數(shù)，第二列的loss

x = data1_loss[:, 0] y = data1_loss[:, 1] x1 = data2_loss[:, 0] y1 = data2_loss[:, 1]

3.先創(chuàng)建一幅圖，再在這幅圖上添加一個(gè)小圖

fig = plt.figure(figsize=(7, 5)) ax1 = fig.add_subplot(1, 1, 1) # ax1是子圖的名字

4.畫出整體的loss曲線

# "g"代表green，表示畫出的曲線是綠色，"-"表示畫出的曲線是實(shí)線，label表示圖例的名稱 plt.plot(x, y, 'r-', label=u'Dense_Unet') plt.plot(x1, y1, 'b-', label=u'RCSCA_Net') plt.legend()plt.xlabel(u'iters') plt.ylabel(u'loss') plt.title('Compare loss for different models in training.') plt.show()

總結(jié)

以上是生活随笔為你收集整理的【深度学习】通过python画出loss曲线的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。