Python Matplotlib 绘制训练集和测试集损失曲线

plt.rcParams['font.family'] = 'SimHei'  # 设置字体为 SimHei，以便正确显示中文
plt.rcParams['axes.unicode_minus'] = False  # 设置负号正常显示

# 从训练记录中获取损失和验证损失数据
acc = history.history['loss']
val_acc = history.history['val_loss']

# 创建横坐标数据
epochs = range(1,len(acc)+1)

# 绘图
plt.figure()
plt.plot(epochs,acc,'bo',label='训练集损失')          # 绘制训练集损失曲线
plt.plot(epochs,val_acc,'b',label='测试集损失')       # 绘制测试集损失曲线
plt.title('训练集测试集损失曲线')                     # 设置标题
plt.legend()      # 添加图例
plt.xlabel('轮次')     # 添加横坐标标签
plt.ylabel('损失')     # 添加纵坐标标签
plt.show()