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.
|
TensorFlow 2 version
View source on GitHub