tf.keras.models.load_model

tf.keras.models.load_model(
    filepath,
    custom_objects=None,
    compile=True
)

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

Loads a model saved via save_model.

Arguments:

• filepath: String, path to the saved model.
• custom_objects: Optional dictionary mapping names (strings) to custom classes or functions to be considered during deserialization.
• compile: Boolean, whether to compile the model after loading.

Returns:

A Keras model instance. If an optimizer was found as part of the saved model, the model is already compiled. Otherwise, the model is uncompiled and a warning will be displayed. When compile is set to False, the compilation is omitted without any warning.

Raises:

• ImportError: if h5py is not available.
• ValueError: In case of an invalid savefile.