tf.keras.models.load_model
View source on GitHub |
Loads a model saved via save_model
.
tf.keras.models.load_model( filepath, custom_objects=None, compile=True )
Arguments | |
---|---|
filepath |
One of the following:
|
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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/models/load_model