Как я могу использовать обученную модель Keras, сохраненную в файле HDF5, чтобы делать прогнозы?

Question

Я недавно начал работать с нейронными сетями. Я построил модель предсказания рукописных символов, используя расширенный набор данных MNIST, sklearn, Keras, numpy и pandas. Основная цель состоит в том, чтобы делать и / или загружать изображения рукописного текста / символов, и модель должна уметь угадывать.

После завершения фазы обучения модель была сохранена в файле my_model.h5. На этом этапе, как я мог бы использовать эту обученную модель (в частности, my_model.h5) в программе Python, которая получает в качестве входных изображений и должна давать прогноз?

Answer 1

Существуют различные способы загрузки (обученной) модели из файла в Keras и TensorFlow.

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

from keras.models import load_model

# returns a compiled model
# identical to the previous one
model = load_model('my_model.h5')

После загрузки модели вы можете использовать model.predict.

Вы интегрируете этот код в свое приложение, когда будете sh.