есть модель 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
Подскажите пожалуйста, как это исправить
