Почему keras жалуется на функцию потерь?

@Andrei_ · Регистрация: 04.08.2020

Author24 — интернет-сервис помощи студентам

Доброго времени суток, есть модель keras, которая сохраняется через model.save() В этой модели используется самописная функция потерь:

Python

1
2
3

def contrastive_loss(y_true, y_pred):
        return K.mean(y_true * K.square(y_pred) + (1 - y_true) * K.square(K.maximum(1 - y_pred, 0)))
model.compile(loss=contrastive_loss, optimizer=rms)

При обучении модели никаких проблем не возникает, но при загрузке модели:

Python

1 2	from tensorflow.keras.models import load_model model = load_model("model.h5")

Летит ошибка: ValueError: Unknown loss function:contrastive_loss
Подскажите пожалуйста, как это исправить

@Nick07 · 07.02.2021, 22:28

Первый запрос GOOGLE выдает решение, Вы его пробовали?

model = load_model('my_model.h5', custom_objects={'my_custom_func':
my_custom_func})

@Andrei_ · 07.02.2021, 22:33 **[ТС]**

Какой запрос вы ввели, напишите пожалуйста
Видимо я гуглить не умею

@Nick07 · 08.02.2021, 10:29

Сообщение от Andrei_

Какой запрос вы ввели

Этот:

Сообщение от Andrei_

Unknown loss function:contrastive_loss

@Nick07 · 09.02.2021, 18:06

4. Сохранение всей модели с индивидуальной метрикой / потерями.
https://medium.com/swlh/saving... 0092fff335

@Nick07 533 / 438 / 47 Регистрация: 17.07.2013 Сообщений: 2,236
	07.02.2021, 22:28	2
	Сообщение было отмечено Andrei_ как решение Решение Первый запрос GOOGLE выдает решение, Вы его пробовали? model = load_model('my_model.h5', custom_objects={'my_custom_func': my_custom_func}) 0

@Andrei_ 0 / 0 / 0 Регистрация: 04.08.2020 Сообщений: 231
	07.02.2021, 22:33 [ТС]	3
	Какой запрос вы ввели, напишите пожалуйста Видимо я гуглить не умею 0

@Nick07 533 / 438 / 47 Регистрация: 17.07.2013 Сообщений: 2,236
	08.02.2021, 10:29	4
	Сообщение от Andrei_ Какой запрос вы ввели Этот: Сообщение от Andrei_ Unknown loss function:contrastive_loss 0

@Nick07 533 / 438 / 47 Регистрация: 17.07.2013 Сообщений: 2,236
	09.02.2021, 18:06	5
	4. Сохранение всей модели с индивидуальной метрикой / потерями. https://medium.com/swlh/saving... 0092fff335 1

Почему keras жалуется на функцию потерь?

Решение