tf.keras.models.save_model

Saves a model as a TensorFlow SavedModel or HDF5 file.

Used in the notebooks

Used in the guide Used in the tutorials

See the Serialization and Saving guide for details.

Usage:

model = tf.keras.Sequential([
    tf.keras.layers.Dense(5, input_shape=(3,)),
    tf.keras.layers.Softmax()])
model.save('/tmp/model')