Save the date! Google I/O returns May 18-20 Register now

tf.keras.models.load_model

TensorFlow 2 version View source on GitHub

Loads a model saved via save_model.

filepath One of the following:

  • String, path to the saved model
  • h5py.File object from which to load the 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.

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.

ImportError if loading from an hdf5 file and h5py is not available.
IOError In case of an invalid savefile.