关于 Keras 模型
这些模型有许多共同的方法和属性:
- 是包含模型网络层的展平列表。
model.inputs
是模型输入张量的列表。model.outputs
是模型输出张量的列表。model.get_config()
返回包含模型配置信息的字典。通过以下代码,就可以根据这些配置信息重新实例化模型:
model.get_weights()
返回模型中所有权重张量的列表,类型为 Numpy 数组。model.set_weights(weights)
从 Numpy 数组中为模型设置权重。列表中的数组必须与get_weights()
返回的权重具有相同的尺寸。model.to_json()
以 JSON 字符串的形式返回模型的表示。请注意,该表示不包括权重,仅包含结构。你可以通过以下方式从 JSON 字符串重新实例化同一模型(使用重新初始化的权重):
- 将模型权重存储为 HDF5 文件。
model.load_weights(filepath, by_name=False)
: 从 HDF5 文件(由save_weights
创建)中加载权重。默认情况下,模型的结构应该是不变的。 如果想将权重载入不同的模型(部分层相同), 设置by_name=True
来载入那些名字相同的层的权重。
除了这两类模型之外,你还可以通过继承 Model
类并在 call
方法中实现你自己的前向传播,以创建你自己的完全定制化的模型,(Model
类继承 API 引入于 Keras 2.2.0)。
网络层定义在 init(self, …)
中,前向传播在 中指定。在 call
中,你可以指定自定义的损失函数,通过调用 self.add_loss(loss_tensor)
(就像你在自定义层中一样)。
model.inputs
和model.outputs
。model.to_yaml()
和model.to_json()
。
关键点:为每个任务使用正确的 API。Model
类继承 API 可以为实现复杂模型提供更大的灵活性,但它需要付出代价(比如缺失的特性):它更冗长,更复杂,并且有更多的用户错误机会。如果可能的话,尽可能使用函数式 API,这对用户更友好。