Модель Keras прогнозируется после сохранения и загрузки - PullRequest
0 голосов
/ 16 января 2020

Я обучил нейронную сеть с помощью Google Colab.
Я сохранил нейронную сеть, используя joblib.dump()

Затем я загрузил модель на свой P C, используя joblib.load()

Я сделал прогноз на одном и том же образце, используя ту же модель, и на colab, и на моем P C. На colab он имеет вывод [[0.51]]. На моем p c он имеет вывод [[nan]].

Сводка модели сообщает, что архитектура модели такая же. Я проверил вес модели, которую я загрузил на свой P C, и модель на colab, и весы точно такие же.

Есть какие-нибудь идеи относительно того, что я могу сделать? Спасибо.

Быстрое обновление: даже если я изменю все свои входные данные на ноль, прогноз будет все еще nan.

1 Ответ

0 голосов
/ 16 января 2020

Насколько я знаю, keras имеет свою собственную функцию для сохранения модели, такую ​​как model.save('file.h5'), а библиотека joblib используется для сохранения моделей sklearn.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...