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

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

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


python machine-learning keras prediction
person Thorvald Olavsen    schedule 06.03.2020    source источник

Ответы (1)


arrow_upward
7
arrow_downward

Есть разные способы загрузки (обученной) модели из файла в 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.

Вы интегрируете этот код в свое приложение по своему усмотрению.

person nbro    schedule 07.03.2020