机器学习-训练模型的保存与恢复(sklearn)
2015-07-31 20:52
302 查看
在做模型训练的时候,尤其是在训练集上做交叉验证,通常想要将模型保存下来,然后放到独立的测试集上测试,下面介绍的是Python中训练模型的保存和再使用。
scikit-learn已经有了模型持久化的操作,导入joblib即可
通过joblib的dump可以将模型保存到本地,clf是训练的分类器
通过joblib的load方法,加载保存的模型。
然后就可以在测试集上测试了
scikit-learn已经有了模型持久化的操作,导入joblib即可
from sklearn.externals import joblib
模型保存
>>> os.chdir("workspace/model_save") >>> from sklearn import svm >>> X = [[0, 0], [1, 1]] >>> y = [0, 1] >>> clf = svm.SVC() >>> clf.fit(X, y) >>> clf.fit(train_X,train_y) >>> joblib.dump(clf, "train_model.m")
通过joblib的dump可以将模型保存到本地,clf是训练的分类器
模型从本地调回
>>> clf = joblib.load("train_model.m")
通过joblib的load方法,加载保存的模型。
然后就可以在测试集上测试了
clf.predit(test_X,test_y)
相关文章推荐
- hadoop之定制自己的sort过程
- (转)女生应该找一个玩ACM的男生
- 【zTree系列】树形结构zTree
- HDOJ1166 敌兵布阵 树状数组
- 喷水装置(一)
- CSS核心技术总结
- UVA 673(括号配对)
- 黑马程序员-C语言文件操作相关函数
- hadoop之定制自己的Partitioner
- 关于 self 和 super 在oc 中 的疑惑 与 分析
- HDU 5313 Bipartite Graph(二分图 + bitset)
- hdoj 4421 Bit Magic 【2-sat 经典题目】
- Android API Guides 阅读笔记(9)----Bound Service
- 【软工】——软工视频总结
- Win7与Ubuntu双系统时卸载Ubuntu的方法
- 秦柯视频与文档资料-全集
- 黑马程序员--C语言——预处理命令
- HUAS Summer Trainning #3~L
- ios-应用管理 字典转模型
- MySQL使用mysqlimport命令导入文本文件