plt将tensorboard几张loss图合为一张的代码

最新推荐文章于 2024-08-19 12:04:49 发布

cv_lz

最新推荐文章于 2024-08-19 12:04:49 发布

阅读量2.8k

点赞数 8

CC 4.0 BY-SA版权

分类专栏： code 文章标签： python 深度学习

本文链接：https://blog.youkuaiyun.com/qq_42250789/article/details/110680109

code 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个使用Python的Matplotlib和NumPy库从CSV文件中读取数据并绘制损失函数图表的方法。通过定义一个读取CSV文件的函数，该程序能够处理特定格式的数据，并将训练次数与对应的损失值用图表展示出来。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
from matplotlib.font_manager import FontProperties
import csv

'''读取csv文件'''

def readcsv(files):
    csvfile = open(files, 'r')
    plots = csv.reader(csvfile, delimiter=',')
    x = []
    y = []
    for row in plots:
        y.append(float(row[2]))
        x.append(float(row[1]))
    return x, y

mpl.rcParams['font.family'] = 'sans-serif'
mpl.rcParams['font.sans-serif'] = 'NSimSun,Times New Roman'

plt.figure()
x2, y2 = readcsv("./g_loss.csv")
plt.plot(x2, y2, color='darkorange', label='g_loss')
# plt.plot(x2, y2, '.', color='red')

# x, y = readcsv("dloss.csv")
# plt.plot(x, y, 'g', label='Without BN')
#
# x1, y1 = readcsv("gloss.csv")
# plt.plot(x1, y1, color='black', label='Without DW and PW')

x4, y4 = readcsv("./d_loss.csv")
plt.plot(x4, y4, color='mediumblue', label='d_loss', linestyle='dashed')

plt.xticks(fontsize=16)
plt.yticks(fontsize=16)

plt.ylim(0, 40)
plt.xlim(0, 18000)


#设置横纵坐标的名称以及对应字体格式
font2 = {'family' : 'Times New Roman',
'weight' : 'normal',
'size'   : 20,
}
font_set = FontProperties(fname=r"c:\windows\fonts\simsun.ttc", size=20)    # matplotlib内无中文字节码，需要自行手动添加
plt.xlabel('训练次数',  font2, fontproperties=font_set)
plt.ylabel('loss值',  font2, fontproperties=font_set)

#设置图例并且设置图例的字体及大小
font1 = {'family' : 'Times New Roman',
'weight' : 'normal',
'size'   : 23,
}
plt.legend(fontsize=16, prop=font1)

# 设置汉字显示正确
# plt.rcParams['font.sans-serif'] = ['SimHei']
# plt.rcParams['axes.unicode_minus'] = False
# 设置保存完全的图
plt.tight_layout()
plt.savefig('loss.png')
plt.show()

结果图：