View source on GitHub |
Parses a yaml model configuration file and returns a model instance.
tf.keras.models.model_from_yaml(
yaml_string, custom_objects=None
)
yaml_string
: YAML string encoding a model configuration.custom_objects
: Optional dictionary mapping names
(strings) to custom classes or functions to be
considered during deserialization.A Keras model instance (uncompiled).
ImportError
: if yaml module is not found.