tf-keras-skeleton.py 文件源码

def model_from_yaml(yaml_string, custom_objects=None):
      """Parses a yaml model configuration file and returns a model instance.

      Arguments:
          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.

      Returns:
          A Keras model instance (uncompiled).

      Raises:
          ImportError: if yaml module is not found.
      """
      if yaml is None:
        raise ImportError('Requires yaml module installed.')
      config = yaml.load(yaml_string)
      return layer_module.deserialize(config, custom_objects=custom_objects)