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