当损失函数是自定义函数时，绘制周期内的 MSE

Question

我使用自定义的损失函数，并希望绘制纪元内的MSE(我使用Keras库(。这是我用来适应我的神经网络并保存历史记录的代码。

model.compile(loss =new_loss2, metrics=['mse'], optimizer=opt)
hist = model3.fit(X_train, y_train, batch_size=32, shuffle=False, epochs=epochs, validation_split=0.15, callbacks = callbackz)

当我尝试使用绘图函数绘制 MSE 时

plt.plot(hist.history['mse'])

我收到此错误

Traceback (most recent call last):
File "D:Keras1 (1).py", line 150, in <module>
plt.plot(hist.history['mse'])
KeyError: 'mse'

我怎样才能以正确的方式绘制它？

Answer 1

不知何故，指标在history字典中通过其扩展名称引用。hist.history['mean_squared_error']应该有效。

注意：如果名称差异困扰您，则在编译模型时也可以使用扩展名称，即model.compile(loss=new_loss2, metrics=['mean_squared_error'], ...).