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

Я использую настраиваемую функцию потерь и хотел бы построить 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'

Как я могу построить это правильно?


keras mse loss-function loss
person Hajar Elhammouti    schedule 28.05.2018    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Каким-то образом на метрики ссылаются по их расширенным именам в словаре history. hist.history['mean_squared_error'] должно работать.

Примечание: если вас смущает несоответствие имени, то при составлении модели можно использовать и расширенное имя, т.е. model.compile(loss=new_loss2, metrics=['mean_squared_error'], ...).

person benjaminplanche    schedule 28.05.2018