tf.keras.models.save_model | TensorFlow v2.16.1 (original) (raw)

tf.keras.models.save_model

Stay organized with collections Save and categorize content based on your preferences.

Saves a model as a .keras file.

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

Used in the notebooks

Used in the tutorials
Train and serve a TensorFlow model with TensorFlow Serving

Args
model	Keras model instance to be saved.
filepath	str or pathlib.Path object. Path where to save the model.
overwrite	Whether we should overwrite any existing model at the target location, or instead ask the user via an interactive prompt.

Example:

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))

Note that model.save() is an alias for keras.saving.save_model().

The saved .keras file contains:

The model's configuration (architecture)
The model's weights
The model's optimizer's state (if any)

Thus models can be reinstantiated in the exact same state.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-06-07 UTC.