此页面由 Cloud Translation API 翻译。
Switch to English

tf.keras.models.model_from_yaml

TensorFlow 1版 GitHub上查看源代码

解析YAML模型配置文件并返回一个模型实例。

用法:

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

yaml_string YAML字符串或打开文件编码模型配置。
custom_objects 可选字典映射名称(字符串),以自定义类或函数来反序列化过程中加以考虑。

一个Keras模型实例(未编译)。

ImportError 如果YAML模块找不到。