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. |