tf.keras.models.save_model

Сохраняет модель в файл .keras.

tf.keras.models.save_model(
    model, filepath, overwrite=True, **kwargs
)

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

Используется в обучающих материалах
Обучение и обслуживание модели TensorFlow с помощью TensorFlow Serving

Аргументы
`model`	Экземпляр модели Keras, которую нужно сохранить.
`filepath`	Объект `str` или `pathlib.Path`. Путь для сохранения модели.
`overwrite`	Следует ли перезаписать существующую модель в целевом расположении или вместо этого запросить подтверждение у пользователя в интерактивном режиме.

Пример:

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 = keras.random.uniform((10, 3))
assert np.allclose(model.predict(x), loaded_model.predict(x))

Обратите внимание, что model.save() — это псевдоним для keras.saving.save_model().

Сохранённый файл .keras содержит:

Конфигурацию модели (архитектуру)
Веса модели
Состояние оптимизатора модели (если есть)

Таким образом, модели можно повторно создать в точно таком же состоянии.

© 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/save_model