@Andrei1penguin1

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

есть модель keras, которая сохраняется через model.save() В этой модели используется самописная функция потерь:
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)

При обучении модели никаких проблем не возникает, но при загрузке модели:
from tensorflow.keras.models import load_model
model = load_model("model.h5")

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

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы