模型保存
生成文件类型是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")