常用 TensorBoard 来显示神经网络训练损失值变化曲线,但需要下载包,操作相对比较麻烦,因此对于非特殊情况可以使用 Matplotlib 来输出曲线,具体代码如下。
import matplotlib.pyplot as plt# 设置汉字输出plt.rcParams['font.sans-serif'] = ['SimSun']plt.rcParams['axes.unicode_minus'] = FalseLoss_list = [] # 损失值存储数组loss = lossFunction(output, target) # 计算损失值,lossFunction自定义Loss_list.append(loss) # 插入损失值plt.figure()x = range(0,1000) # x和y的维度要一样y = Loss_listplt.plot(x, y, 'r-') # 设置输出样式plt.ylabel('当前损失值')plt.xlabel('训练次数/次')plt.savefig('F://loss.jpg') # 保存训练损失曲线图片plt.show() # 显示曲线