百度360必应搜狗淘宝本站头条
python类型转换python串口编程python计时python异常python进度条
当前位置：网站首页 > 技术资源 > 正文

python机器学习:如何储存训练好的模型并重新调用

off999 2024-11-22 19:02 23 浏览 0 评论

一篇文章写清楚一个问题，关注我，自学python！

今天用一个实例给大家写写在机器学习中如何将训练好的模型储存起来方便下次调用。

模型储存的必要性

当我们训练好一个机器学习模型后，如果下次还想用这个模型，我们就需要把这个模型保存下来，尤其是在大数据中训练模型很耗时间的情况下，一定要保存模型，然后下次直接导入就好了，不然每次都跑一遍，训练时间短还好说，要是一次跑好几天的那种模型，不保存就会很悲催。

今天给大家介绍两种保存机器学习模型的方法：


  • 一种就是使用python自带的pickle
  • 另一种使用sklearn中的模块joblib

pickle储存模型实际操练

首先我们看pickle怎么用，要储存的模型还是我上一篇文章训练好的多元线性回归模型，见python机器学习：多元线性回归模型实战，现在我们将模型储存起来，代码如下

import pickle
with open('reg_pickle','wb') as f:
    pickle.dump(reg,f)

导入pickle后，用写入模式打开一个reg_pickle文件，然后将我们的模型reg存入到f中。

运行代码就可以看到我们的工作目录中出现了一个新的文件，这个新文件就是我们储存的模型


pickle调用模型

调用模型代码如下：

with open('reg_pickle','rb') as f:
    mp = pickle.load(f)

用只读模式打开我们刚刚储存的reg_pickle文件，然后用.load方法就可以载入模型再次使用啦

joblib储存模型

现在来看joblib的用法，要储存的模型还是我上一篇文章训练好的多元线性回归模型，见python机器学习：多元线性回归模型实战，现在我们将模型储存起来，代码如下：

from sklearn.externals import joblib
joblib.dump(reg, 'reg_joblib')

首先载入joblib库，然后直接用.dump方法就可以实现对模型的储存，比第一种方法要简单得多。

运行代码就可以看见工作目录中出现了reg_joblib文件：


joblib调用模型

调用模型很简单，只需要用.load方法就行

mj = joblib.load('reg_joblib')

可以看到模型系数和原来模型一模一样。

小结

今天给大家介绍了两种储存机器学习模型的方法，相比pickle，我更加推荐大家使用joblib，因为更简洁，还有根据官方文档介绍joblib效率也更高。感谢大家耐心看完。发表这些东西的主要目的就是督促自己，希望大家关注评论指出不足，一起进步。内容我都会写的很细，用到的数据集也会在原文中给出链接，你只要按照文章中的代码自己也可以做出一样的结果，一个目的就是零基础也能懂，因为自己就是什么基础没有从零学Python的，加油。

（站外链接发不了，请关注后私信回复“数据链接”获取本头条号所有使用数据）

往期内容：

python机器学习：多元线性回归模型实战

python机器学习：如何划分训练集和测试集

相关推荐

阿里云国际站ECS:阿里云ECS如何提高网站的访问速度?

TG：@yunlaoda360引言：速度即体验，速度即业务在当今数字化的世界中，网站的访问速度已成为决定用户体验、用户留存乃至业务转化率的关键因素。页面加载每延迟一秒，都可能导致用户流失和收入损失。对...

高流量大并发Linux TCP性能调优_linux 高并发网络编程

其实主要是手里面的跑openvpn服务器。因为并没有明文禁p2p（哎……想想那么多流量好像不跑点p2p也跑不完），所以造成有的时候如果有比较多人跑BT的话，会造成VPN速度急剧下降。本文所面对的情况为...

性能测试100集(12)性能指标资源使用率

在性能测试中，资源使用率是评估系统硬件效率的关键指标，主要包括以下四类：#性能测试##性能压测策略##软件测试#1.CPU使用率定义：CPU处理任务的时间占比，计算公式为1-空闲时间/总...

Linux 服务器常见的性能调优_linux高性能服务端编程

一、Linux服务器性能调优第一步——先搞懂“看什么”很多人刚接触Linux性能调优时，总想着直接改配置，其实第一步该是“看清楚问题”。就像医生看病要先听诊，调优前得先知道服务器“哪里...

Nginx性能优化实战:手把手教你提升10倍性能!

关注△mikechen△，十余年BAT架构经验倾囊相授！Nginx是大型架构而核心，下面我重点详解Nginx性能@mikechen文章来源：mikechen.cc1.worker_processe...

高并发场景下，Spring Cloud Gateway如何抗住百万QPS?

关注△mikechen△，十余年BAT架构经验倾囊相授！大家好，我是mikechen。高并发场景下网关作为流量的入口非常重要，下面我重点详解SpringCloudGateway如何抗住百万性能@m...

Kubernetes 高并发处理实战(可落地案例 + 源码)

目标场景：对外提供HTTPAPI的微服务在短时间内收到大量请求（例如每秒数千至数万RPS），要求系统可弹性扩容、限流降级、缓存减压、稳定运行并能自动恢复。总体思路（多层防护）：边缘层：云LB...

高并发场景下，Nginx如何扛住千万级请求?

Nginx是大型架构的必备中间件，下面我重点详解Nginx如何实现高并发@mikechen文章来源：mikechen.cc事件驱动模型Nginx采用事件驱动模型，这是Nginx高并发性能的基石。传统...

Spring Boot+Vue全栈开发实战，中文版高清PDF资源

SpringBoot+Vue全栈开发实战，中文高清PDF资源，需要的可以私我：）SpringBoot致力于简化开发配置并为企业级开发提供一系列非业务性功能，而Vue则采用数据驱动视图的方式将程序...

Docker-基础操作_docker基础实战教程二

一、镜像1、从仓库获取镜像搜索镜像：dockersearchimage_name搜索结果过滤：是否官方:dockersearch--filter="is-offical=true...

你有空吗?跟我一起搭个服务器好不好?

来人人都是产品经理【起点学院】，BAT实战派产品总监手把手系统带你学产品、学运营。昨天闲的没事的时候，随手翻了翻写过的文章，发现一个很严重的问题。就是大多数时间我都在滔滔不绝的讲理论，却很少有涉及动手...

部署你自己的 SaaS_saas如何部署

部署你自己的VPNOpenVPN——功能齐全的开源VPN解决方案。（DigitalOcean教程）dockovpn.io—无状态OpenVPNdockerized服务器，不需要持久存储。...

Docker Compose_dockercompose安装

DockerCompose概述DockerCompose是一个用来定义和管理多容器应用的工具，通过一个docker-compose.yml文件，用YAML格式描述服务、网络、卷等内容，...

京东T7架构师推出的电子版SpringBoot，从构建小系统到架构大系统

前言：Java的各种开发框架发展了很多年，影响了一代又一代的程序员，现在无论是程序员，还是架构师，使用这些开发框架都面临着两方面的挑战。一方面是要快速开发出系统，这就要求使用的开发框架尽量简单，无论...

Kubernetes (k8s) 入门学习指南_k8s kubeproxy

Kubernetes(k8s)入门学习指南一、什么是Kubernetes？为什么需要它？Kubernetes(k8s)是一个开源的容器编排系统，用于自动化部署、扩展和管理容器化应用程序。它...

取消回复欢迎 你 发表评论:

一周热门
最近发表
标签列表
Copyright Your WebSite.Some Rights Reserved.