tf.keras.models.load_model
Loads a model saved via save_model
.
tf.keras.models.load_model(
filepath, custom_objects=None, compile=True
)
Arguments |
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. |
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 loading from an hdf5 file and h5py is not available. |
IOError | In case of an invalid savefile. |