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.