sklearn模型保存与加载
2024-08-30 13:42:46
sklearn模型保存与加载
sklearn模型的保存和加载API
from sklearn.externals import joblib
# 保存:joblib.dump(estimator, 'test.pkl')
# 加载:estimator = joblib.load('test.pkl')
注意:保存的后缀名是.pkl
线性回归的模型保存加载案例
保存模型
# 1.获取数据
data = load_boston()
# 2.数据集划分 数据的基本处理
x_train, x_test, y_train, y_test = train_test_split(data.data, data.target, random_state=22)
# 3.特征工程-标准化
transfer = StandardScaler()
x_train = transfer.fit_transform(x_train)
x_test = transfer.fit_transform(x_test)
# 4.1 创建模型 实例化估计器
estimator = Ridge(alpha=1)
# 4.2 训练模型 fit 正规方程计算得到最优可训练参数
estimator.fit(x_train, y_train)
#保存模型
joblib.dump(estimator,'./test.pkl')
# 加载模型
estimator = joblib.load('./test.pkl')
# 5.模型评估
y_predict = estimator.predict(x_test)
print("预测值为:\n", y_predict)
print("模型中的系数为:\n", estimator.coef_)
print("模型中的偏置为:\n", estimator.intercept_)
error = mean_squared_error(y_test, y_predict)
print("误差为:\n", error)
最新文章
- PHP Date ( I need to use)
- 系列文章:老项目的#iPhone6与iPhone6Plus适配#(持续更新中,更新日期2014年10月12日 星期日 )
- virtualenv and virtualenvwrapper on Ubuntu 14.04
- a标签属性说明
- 【原创】StickHeaderListView的简单实现,解决footerView问题
- ActiveReports 交互式报表之向下钻取解决方案
- vertical-align:middle的居中细节调整
- 弹飞DZY(思维,打表,还没过全,先放着)
- LightOJ 1234 Harmonic Number 调和级数部分和
- Angular 学习笔记 ( CDK - Portal )
- keras04 GAN simple
- WPF布局的应用
- vscode常用快捷键和插件(持续更新),以及一些常用设置的坑和技巧
- 使用DigitalOcean控制台访问Droplet(远程服务器)
- Django的缓存
- C# 中 DataTable转换成IList
- 修改无线wifi网络名称。注册表。windows 无线属性 windows 无线 配置文件
- Canvas+Js制作动量守恒的小球碰撞
- NSOperation的使用细节 [2]
- [CF1042D] Petya and Array
热门文章
- redis的安装与设置开机自启动
- virtualbox linux客户机中安装增强功能包缺少kernel头文件问题解决
- Java面试总结 Boss沟通过:500+,面试:20,已投简历130+
- 鸿蒙内核源码分析(自旋锁篇) | 当立贞节牌坊的好同志 | 百篇博客分析OpenHarmony源码 | v26.02
- (目录)Fortran学习笔记:开坑!!!
- .Net Core with 微服务 - 使用 AgileDT 快速实现基于可靠消息的分布式事务
- Initialize this repository with a README
- 专业网络损伤仪HoloWAN meme只需5999元!
- js--typeof 和 instanceof 判断数据类型的区别及开发中的使用
- 轻量级 Java 基础开发框架,Solon &; Solon Cloud 1.5.48 发布