keras模型保存、加载

模型保存

生成文件类型是HDF5，需要安装h5py。安装命令：pip install h5py

model.save("xxxxx.hdf5")  # 文件名  可以指定文件路径

模型加载

首先需要导入依赖：from keras.models import load_model

load_model("mnist_best.hdf5")  # 加载模型

保持和加载有只保存参数和网络结构两种
仅保存参数模型

model.save_weights("xxxxx.hdf5")

仅加载参数模型

model.load_weights("xxxxx.hdf5")