Arhn - архитектура программирования

Как загрузить модель Keras с пользовательской функцией потерь?

Я создал следующую пользовательскую функцию потерь:

RMSE = function(y_true,y_pred) {
         k_sqrt(k_mean(k_square(y_pred - y_true))) 
    }

И он отлично работает, когда я сохранил модель. Однако, когда я снова загрузил модель, используя:

load_model_hdf5(filepath= "modelpath")

Я получаю следующую ошибку:

#Error in py_call_impl(callable, dots$args, dots$keywords):
#      valueError: Unknown loss function:RMSE

Возможно, этот вопрос имеет что-то общее с этим one Я сделал раньше. Что мне делать, чтобы перестать получать эту ошибку?

r keras tensorflow loss-function
09.03.2020

  • Попробуйте использовать load_model_hdf5 (путь к файлу = путь к модели, компиляция = ложь). Что вы получаете? Не могли бы вы также включить остальную часть вашей модели? 10.03.2020

Ответы:


1

Поскольку вы используете в своей модели пользовательскую функцию потерь, функция потерь не будет сохранена при сохранении модели на диске, а вместо этого в файл модели будет включено только ее имя. Затем, когда вы хотите загрузить модель позже, вам нужно сообщить модели о соответствующей функции потерь для сохраненного имени. Чтобы обеспечить это сопоставление, вы можете использовать аргумент custom_objects функции load_model_hdf5:

load_model_hdf5(filepath = "modelpath", custom_objects = list(RMSE = RMSE))

В качестве альтернативы, после завершения обучения, если вы просто хотите использовать модель для прогнозирования, вы можете просто передать аргумент compile = False функции load_model_hdf5 (следовательно, функция потерь не потребуется и не будет загружена):

load_model_hdf5(filepath = "modelpath", compile = False)
11.04.2020
Новые материалы
Коллекции публикаций по глубокому обучению
Последние пару месяцев я создавал коллекции последних академических публикаций по различным подполям глубокого обучения в моем блоге https://amundtveit.com - эта публикация дает обзор 25..
Представляем: Pepita
Фреймворк JavaScript с открытым исходным кодом Я знаю, что недостатка в фреймворках JavaScript нет. Но я просто не мог остановиться. Я хотел написать что-то сам, со своими собственными..
Советы по коду Laravel #2
1-) Найти // You can specify the columns you need // in when you use the find method on a model User::find(‘id’, [‘email’,’name’]); // You can increment or decrement // a field in..
Работа с временными рядами спутниковых изображений, часть 3 (аналитика данных)
Анализ временных рядов спутниковых изображений для данных наблюдений за большой Землей (arXiv) Автор: Рольф Симоэс , Жильберто Камара , Жильберто Кейрос , Фелипе Соуза , Педро Р. Андраде ,..
3 способа решить квадратное уравнение (3-й мой любимый) -
1. Методом факторизации — 2. Используя квадратичную формулу — 3. Заполнив квадрат — Давайте поймем это, решив это простое уравнение: Мы пытаемся сделать LHS,..
Создание VR-миров с A-Frame
Виртуальная реальность (и дополненная реальность) стали главными модными терминами в образовательных технологиях. С недорогими VR-гарнитурами, такими как Google Cardboard , и использованием..
Демистификация рекурсии
КОДЕКС Демистификация рекурсии Упрощенная концепция ошеломляющей О чем весь этот шум? Рекурсия, кажется, единственная тема, от которой у каждого начинающего студента-информатика..