DeepLearning tutorial（2）机器学习算法在训练过程中保存参数

最新推荐文章于 2022-03-29 10:20:12 发布

wepon_

最新推荐文章于 2022-03-29 10:20:12 发布

阅读量1.2w

点赞数 10

分类专栏： Machine Learning 深度学习入门指南文章标签： python deeplearning 保存参数

本文链接：https://blog.csdn.net/u012162613/article/details/43169019

版权

本文介绍在Python中如何读取`.pkl.gz`文件，以及在机器学习算法训练过程中保存参数的方法。通过gzip和cPickle模块读取压缩文件，当验证损失降低时保存最优参数。

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

DeepLearning tutorial（2）机器学习算法在训练过程中保存参数

@author：wepon

用到python里的gzip以及cPickle模块，简单的使用代码如下，如果想详细了解可以参考上面给出的链接。

#以读取mnist.pkl.gz为例
import cPickle, gzip
f = gzip.open('mnist.pkl.gz', 'rb')
train_set, valid_set, test_set = cPickle.load(f)
f.close()

其实就是分两步，先读取gz文件，再读取pkl文件。pkl文件的应用正是下文要讲的，我们用它来保存机器学习算法训练过程中的参数。

关注

专栏目录