Python机器学习手册：从数据预处理到深度学习

问题描述

有一个训练好的Keras模型，你想保存它并在别处加载它。

解决方案

将模型保存为HDF5文件：

然后，就可以在另一个应用或者其他训练中加载这个模型：

讨论

不同于scikit-learn,Keras不推荐使用pickle格式来保存模型，而是要将模型保存为HDF5文件。HDF5文件包含了你需要的一切，不仅包含加载模型做预测所需要的结构和训练后的参数，而且包含重新训练所需要的各种设置（即损失、优化器的设置和当前状态）。