График MSE по эпохам, когда функция потерь - это настраиваемая функция

Question

Я использую настраиваемую функцию потерь и хотел бы построить MSE в эпоху (я использую Keras Library).Это код, который я использую, чтобы соответствовать своей нейронной сети и сохранить историю.

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 с помощью функции plot

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'], ...).