TensorFlow 2 version View source on GitHub

Parses a yaml model configuration file and returns a model instance.

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.