tf.keras.models.model_from_yaml
Parses a yaml model configuration file and returns a model instance.
tf.keras.models.model_from_yaml(
yaml_string, custom_objects=None
)
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. |