tf.keras.models.save_model

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

Defined in tensorflow/python/keras/engine/saving.py.

Saves a model to a HDF5 file.

The saved model contains: - the model's configuration (topology) - the model's weights - the model's optimizer's state (if any)

Thus the saved model can be reinstantiated in the exact same state, without any of the code used for model definition or training.

Arguments:

• model: Keras model instance to be saved.
• filepath: One of the following: - String, path where to save the model - h5py.File object where to save the model
• overwrite: Whether we should overwrite any existing model at the target location, or instead ask the user with a manual prompt.
• include_optimizer: If True, save optimizer's state together.

Raises:

• ImportError: if h5py is not available.