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

D0644727

于 2020-11-15 17:47:04 发布

阅读量1.1k

点赞数 1

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

本文链接：https://blog.youkuaiyun.com/D0644727/article/details/109704936

版权

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

以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元/天解锁文章