1.模型的保存与载入
代码
1.1安装h5py
要保存模型,首先要安装h5py
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple h5py
1.2保存模型
在训练完模型后,保存模型
#训练模型
model.fit(x_train,y_train,batch_size=60,epochs=20)
#保存模型,这里是相对路劲
model.save('save_model.h5')

1.3载入模型
1.3.1载入模型
#导入load_model

本文详细介绍了在Keras中如何保存和载入模型,包括使用h5py库进行模型保存,以及利用ModelCheckpoint回调函数在训练过程中保存最佳模型的方法。通过示例代码展示了如何在训练结束后载入模型进行测试或继续训练。
最低0.47元/天 解锁文章
191

被折叠的 条评论
为什么被折叠?



