常用 TensorBoard 来显示神经网络训练损失值变化曲线,但需要下载包,操作相对比较麻烦,因此对于非特殊情况可以使用 Matplotlib 来输出曲线,具体代码如下。

import matplotlib.pyplot as plt
# 设置汉字输出
plt.rcParams['font.sans-serif'] = ['SimSun']
plt.rcParams['axes.unicode_minus'] = False

Loss_list = [] # 损失值存储数组

loss = lossFunction(output, target) # 计算损失值,lossFunction自定义

Loss_list.append(loss) # 插入损失值

plt.figure()
x = range(0,1000) # x和y的维度要一样
y = Loss_list
plt.plot(x, y, 'r-') # 设置输出样式
plt.ylabel('当前损失值')
plt.xlabel('训练次数/次')
plt.savefig('F://loss.jpg') # 保存训练损失曲线图片
plt.show() # 显示曲线
Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐