tf.keras.models.load_model  |  TensorFlow v1.15.0 (original) (raw)

tf.keras.models.load_model

Stay organized with collections Save and categorize content based on your preferences.

Loads a model saved via save_model.

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.keras.models.load_model, `tf.compat.v2.keras.models.load_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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2020-10-01 UTC.