tf.keras.models.load_model

Загружает модель, сохранённую с помощью model.save().

tf.keras.models.load_model(
    filepath, custom_objects=None, compile=True, safe_mode=True
)

Используется в блокнотах

Используется в руководстве	Используется в учебных пособиях
Вступление к модулям, слоям и моделям Миграция `tf.feature_column` в слои предварительной обработки Keras Миграция рабочего процесса SavedModel Миграция кода TFLite в TF2 Многопроцессорная и распределённая тренировка	Сохранение и загрузка модели с помощью стратегии распределения Сохранение и загрузка моделей Распределённая тренировка с Keras Тренировка с участием нескольких узлов с помощью Keras Перенос обучения с помощью TensorFlow Hub

Аргументы
`filepath`	Объект `str` или путь к сохранённому файлу модели `pathlib.Path`.
`custom_objects`	Необязательный словарь, сопоставляющий имена (строки) с пользовательскими классами или функциями, которые нужно учитывать при десериализации.
`compile`	Булево значение, компилировать ли модель после загрузки.
`safe_mode`	Булево значение, запрещать ли небезопасную десериализацию `lambda`. При `safe_mode=False` загрузка объекта может привести к выполнению произвольного кода. Этот аргумент применимо только к формату модели Keras v3. По умолчанию `True`.

Возвращаемое значение
Экземпляр модели Keras. Если исходная модель была скомпилирована и аргумент `compile=True` задан, то возвращаемая модель будет скомпилирована. В противном случае модель останется нескомпилированной.

Пример:

model = keras.Sequential([
    keras.layers.Dense(5, input_shape=(3,)),
    keras.layers.Softmax()])
model.save("model.keras")
loaded_model = keras.saving.load_model("model.keras")
x = np.random.random((10, 3))
assert np.allclose(model.predict(x), loaded_model.predict(x))

Обратите внимание, что значения переменных модели могут иметь другие имена (свойство var.name, например, "dense_1/kernel:0") после перезагрузки. Рекомендуется использовать атрибуты слоя для доступа к конкретным переменным, например, model.get_layer("dense_1").kernel.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/models/load_model