Простые способы загрузки модели Keras из JSON: пошаговое руководство

Загрузка модели Keras из JSON — обычная задача в проектах глубокого обучения. Это позволяет вам восстановить архитектуру и веса предварительно обученной модели, готовя ее к прогнозированию или дальнейшему обучению. В этом сообщении блога мы рассмотрим несколько методов загрузки модели Keras из файла JSON с использованием примеров кода Python. Итак, давайте углубимся и узнаем, как вернуть к жизни сохраненные модели!

Метод 1: использование функции Keras model_from_json
Функция model_from_jsonв Keras обеспечивает простой способ загрузки модели из файла JSON. Вот пример кода:

from tensorflow import keras
# Load JSON file
with open('model.json', 'r') as json_file:
    loaded_model_json = json_file.read()
# Load model from JSON
loaded_model = keras.models.model_from_json(loaded_model_json)
# Load model weights
loaded_model.load_weights('model_weights.h5')

Метод 2: объединение файлов JSON и HDF5.
Иногда архитектура модели хранится в файле JSON, а веса сохраняются отдельно в файле HDF5. В таких случаях можно использовать следующий код:

from tensorflow import keras
# Load JSON file
with open('model.json', 'r') as json_file:
    loaded_model_json = json_file.read()
# Load model from JSON
loaded_model = keras.models.model_from_json(loaded_model_json)
# Load model weights from HDF5 file
loaded_model.load_weights('model_weights.h5')

Метод 3: использование json.loadи model_from_config
Альтернативный подход — использовать функцию json.loadдля загрузки файла JSON, а затем вручную создайте модель, используя словарь конфигурации. Вот пример:

from tensorflow import keras
import json
# Load JSON file
with open('model.json', 'r') as json_file:
    config = json.load(json_file)
# Create model from configuration
loaded_model = keras.models.model_from_config(config)
# Load model weights from HDF5 file
loaded_model.load_weights('model_weights.h5')

В этом сообщении блога мы рассмотрели различные методы загрузки модели Keras из файла JSON. Мы рассмотрели функцию model_from_json, объединив файлы JSON и HDF5 и используя json.loadс model_from_config. Эти методы позволят вам легко восстановить сохраненные модели Keras. Не стесняйтесь выбирать метод, который лучше всего соответствует вашим потребностям, и быстро начните работу с моделями!