W3cubDocs

/TensorFlow 2.3

tf.keras.models.model_from_yaml

View source on GitHub

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

Usage:

model = tf.keras.Sequential([
    tf.keras.layers.Dense(5, input_shape=(3,)),
    tf.keras.layers.Softmax()])
try:
  import yaml
  config = model.to_yaml()
  loaded_model = tf.keras.models.model_from_yaml(config)
except ImportError:
  pass
Arguments
yaml_string YAML string or open file encoding a model configuration.
custom_objects Optional dictionary mapping names (strings) to custom classes or functions to be considered during deserialization.
Returns
A Keras model instance (uncompiled).
Raises
ImportError if yaml module is not found.

© 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/r2.3/api_docs/python/tf/keras/models/model_from_yaml