如何保存调用机器学习中的训练结果

最新推荐文章于 2023-08-27 20:22:43 发布

D0644727

最新推荐文章于 2023-08-27 20:22:43 发布

阅读量1k

点赞数 1

分类专栏：机器学习文章标签：机器学习

本文链接：https://blog.csdn.net/D0644727/article/details/109704936

版权

本文以mnist数据集为例，探讨如何在机器学习中保存和读取训练结果。通过load_weights方法加载权重，注意ckpt文件可能包含索引信息。同时，讨论了完整保存模型的必要性。

摘要由CSDN通过智能技术生成

以mnist数据集的训练结果保存与调用为例子：

模型读取

load_weights(路径文件名）
一般在形成ckpt文件过程中常常会出现索引，因此可用于判断是否可以直接加载模型。

checkpoint_save_path = "./checkpoint/mnist.ckpt"
if os.path.exists(checkpoint_save_path + '.index'):
    print('-------------load the model-----------------')
    model.load_weights

最低0.47元/天解锁文章

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

D0644727

关注关注

1
点赞
踩
2

收藏

觉得还不错? 一键收藏
0
评论
如何保存调用机器学习中的训练结果

以mnist数据集的训练结果保存与调用为例子：模型读取load_weights(路径文件名）一般在形成ckpt文件过程中常常会出现索引，因此可用于判断是否可以直接加载模型。checkpoint_save_path = "./checkpoint/mnist.ckpt"if os.path.exists(checkpoint_save_path + '.index'): print('-------------load the model-----------------') mod
复制链接

扫一扫