tf.keras.models.load_model

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

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

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

Используется в руководстве	Используется в учебных материалах
Введение в модули, слои и модели Перенос `tf.feature_column`s на слои предварительной обработки Keras Перенос рабочего процесса SavedModel Перенос кода TFLite на TF2 Обучение на нескольких GPU и распределённое обучение	Сохранение и загрузка модели с помощью стратегии распределения Сохранение и загрузка моделей Распределённое обучение с 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