sklearn训练模型的保存与加载
时间:2019-12-03
本文章向大家介绍sklearn训练模型的保存与加载,主要包括sklearn训练模型的保存与加载使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。
使用joblib模块保存于加载模型
在机器学习的过程中,我们会进行模型的训练,最常用的就是sklearn中的库,而对于训练好的模型,我们当然是要进行保存的,不然下次需要进行预测的时候就需要重新再进行训练。如果数据量小的话,那再重新进行训练是没有问题的,但是如果数据量大的话,再重新进行训练可能会花费很多开销,这个时候,保存好已经训练的模型就显得特别重要了。我们可以使用sklearn中的joblib模块进行保存与加载。
from sklearn.externals import joblib # 保存模型 joblib.dump(lr, "./ML/test.pkl") # lr是训练好的模型, "./ML/test.pkl"是模型要保存的路径及保存模型的文件名,其中,'pkl' 是sklearn中默认的保存格式gai
这样,我们就将模型保存好了,当我们需要使用到该模型时,可以直接使用joblib.load进行加载模型
lr = joblib.load("./ML/test.pkl") # 进行模型的预测 y_pred = lr.predict(x_test) # 加载出来的模型跟我们训练出来的模型一样,有相同的参数
原文地址:https://www.cnblogs.com/ceo-python/p/11979978.html
- 系统开发之设计模式
- centos7修改系统语言为简体中文
- vue之导入Bootstrap以及jQuery的两种方式
- SQLAlchemy中的自引用
- centos下安装pip2
- 修改mysql的时间/时区
- windows查看端口占用情况
- 大坑记录 - shell脚本删除操作
- mysql5.7 创建新表时提示时间戳非法
- #测试框架推荐# test4j,数据库测试
- C语言中的封装 - 答读者问
- pip is configured with locations that require TLS/SSL, however the ssl module in Python is not avail
- 为什么SOA如此重要
- Grunt配置及使用
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法
- REST服务,使用Dubbo还是SpringMVC?
- Linux系统下Anaconda的安装和使用教程
- Flutter Dojo设计之道——利用Github打造完善的开源项目
- 最强 Redis 客户端 lettuce 已支持 Redis6客户端
- 还在手动整理数据库文档?试试这个工具
- Elasticsearch 常见的 8 种错误及最佳实践
- Spark流式状态管理
- Scala中的IO操作及ArrayBuffer线程安全问题
- 设计模式之单例模式
- Roslyn 理解 msbuild 的清理过程
- gorm聚合查询group结合join和count
- 潘石屹用Python解决100个问题 | 集合
- Catalina 默认使用zsh了,你可习惯
- LeetCode 354 Russian Doll Envelopes (动态规划)
- 设计模式之原型模式