保存模型

使用model.save()方法可一键保存整个Keras模型,包含架构、权重和优化器状态:

model.save("my_model.h5")

📌 保存格式:

  • .h5:HDF5格式(默认)
  • .tf:TensorFlow格式(支持更复杂的模型)
模型保存流程

加载模型

通过tf.keras.models.load_model()恢复模型:

from tensorflow.keras.models import load_model
model = load_model("my_model.h5")

🔄 加载注意事项:

  1. 需要与保存时相同的架构定义
  2. 可通过custom_objects参数加载自定义层
  3. 推荐使用tf.saved_model.save()进行生产环境保存

模型优化技巧

  • 使用model.save_weights()仅保存权重(节省空间)
  • 通过tf.keras.utils.serialize_keras_object()实现自定义序列化
  • 了解更多信息,请访问本站的模型优化指南
模型加载示例