Keras load_model с пользовательскими объектами не работает должным образом

Question

Настройка

Как уже упоминалось в заголовке, у меня возникла проблема с моей пользовательской функцией потерь при попытке загрузить сохраненную модель. Моя потеря выглядит следующим образом:

def weighted_cross_entropy(weights):

    weights = K.variable(weights)

    def loss(y_true, y_pred):
        y_pred = K.clip(y_pred, K.epsilon(), 1-K.epsilon())

        loss = y_true * K.log(y_pred) * weights
        loss = -K.sum(loss, -1)
        return loss

    return loss

weightes_loss = weighted_cross_entropy([0.1,0.9])

Итак, во время тренировок я использовал функцию weighted_loss как функцию потерь, и все работало хорошо. По окончании обучения я сохраняю модель в виде файла .h5 со стандартной функцией model.save из keras API.

Задача

Когда я пытаюсь загрузить модель через

model = load_model(path,custom_objects={"weighted_loss":weighted_loss})

Я получаю ValueError о том, что потеря неизвестна.

Error

Сообщение об ошибке выглядит следующим образом:

File "...\predict.py", line 29, in my_script
"weighted_loss": weighted_loss})
File "...\Continuum\anaconda3\envs\processing\lib\site-packages\keras\engine\saving.py", line 419, in load_model
model = _deserialize_model(f, custom_objects, compile)
File "...\Continuum\anaconda3\envs\processing\lib\site-packages\keras\engine\saving.py", line 312, in _deserialize_model
sample_weight_mode=sample_weight_mode)
File "...\Continuum\anaconda3\envs\processing\lib\site-packages\keras\engine\training.py", line 139, in compile
loss_function = losses.get(loss)
File "...\Continuum\anaconda3\envs\processing\lib\site-packages\keras\losses.py", line 133, in get
return deserialize(identifier)
File "...\Continuum\anaconda3\envs\processing\lib\site-packages\keras\losses.py", line 114, in deserialize
printable_module_name='loss function')
File "...\Continuum\anaconda3\envs\processing\lib\site-packages\keras\utils\generic_utils.py", line 165, in deserialize_keras_object
':' + function_name)
ValueError: Unknown loss function:loss

Вопросы

Как я могу исправить эту проблему? Возможно ли, что причиной этого является мое определение потерь? Так что keras не знает, как обращаться с переменной weights?

Answer 1

Название вашей функции потери loss (т.е. def loss(y_true, y_pred):).Поэтому при загрузке модели вам нужно указать 'loss' в качестве имени:

model = load_model(path, custom_objects={'loss': weighted_loss})