机器学习基础15-模型保存

结果部署是机器学习项目中的最后一步,也是最重要的步骤之一。选

定算法之后,对算法训练生成模型,并部署到生产环境上,以便利用机器学习解决实际问题。模型生成之后,也需要定期对模型进行更新,使模型处于最新、最有效的状态,通常建议

3~6个月更新一次模型。

持久化加载模型

找到一个能够生成高准确度模型的算法不是机器学习最后的步骤,在实际的项目中,需要将生成的模型序列化,并将其发布到生产环境。当有新数据出现时,需要反序列化已保存的模型,然后用其预测新的数据。接下来将介绍在Python中如何序列化和反序列化scikit-learn的模型。

本章内容将包括以下几个方面:

  • 模型序列化和重用的重要性。
  • 如何通过pickle来序列化和反序列化机器学习的模型。
  • 如何通过joblib来序列化和反序列化机器学习的模型。

通过pickle序列化和反序列化机器学习的模型

pickle是标准的Python序列化的方法,可以通过它来序列化机器学习算法生成的模型,并将其保存到文件中。当需要对新数据进行预测时,将保存在文件中的模型反序列化,并用其来预测新数据的结果。

下面给出一个根据 Pima Indians数据集训练逻辑回归算法生成的一个模型,并将其序列化到文件,然后反序列化这个模型的例子。在机器学习项目中,当模型训练需要花费大量的时间时,模型序列化是尤为重要的。

代码如下:

cpp 复制代码
import pickle

import pandas as pd

from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split

#数据预处理
path = 'D:\down\\archive\\diabetes.csv'
data = pd.read_csv(path)

#打印标签名称
print(data.columns[0:8])

#将数据转成数组
array = data.values
#分割数据,去掉最后一个标签
X = array[:, 0:8]

Y = array[:, 8]

test_size = 0.33
seed = 4

X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=test_size, random_state=seed)

model = LogisticRegression()

model.fit(X_train, Y_train)

#保存模型
filename = 'finalized_model.sav'

with open(filename, 'wb') as f:

    pickle.dump(model, f)

#加载模型
with open(filename, 'rb') as f:

    #模型反序列化
    loaded_model = pickle.load(f)

    result = loaded_model.score(X_test, Y_test)

    print("算法评估结果:%.3f%%" % (result * 100.0))

运行结果:

cpp 复制代码
算法评估结果:80.709%

同时也会生成模型文件

相关推荐
2501_9248772112 分钟前
强逆光干扰漏检率↓78%!陌讯多模态融合算法在光伏巡检的实战优化
大数据·人工智能·算法·计算机视觉·目标跟踪
算家计算28 分钟前
多模态融合新纪元:Ovis2.5 本地部署教程,实现文本、图像与代码的深度协同推理
人工智能·开源
算家计算31 分钟前
全球AI百强榜发布!中国产品占据移动端半壁江山
人工智能·资讯·deepseek
max50060034 分钟前
北京大学MuMo多模态肿瘤分类模型复现与迁移学习
人工智能·python·机器学习·分类·数据挖掘·迁移学习
2501_9248773536 分钟前
智慧零售漏扫率↓79%!陌讯多模态融合算法在智能收银与货架管理的实战解析
大数据·人工智能·算法·目标检测·边缘计算·零售
后端小肥肠1 小时前
公众号想做 10w + 趣味漫画还在卡壳?Coze 工作流从 0 到 1 教你搭,难点全拆 + 一键出稿
人工智能·aigc·coze
xz2024102****1 小时前
吴恩达机器学习作业二:线性可分逻辑回归
人工智能·机器学习·逻辑回归
机器之心1 小时前
AAAI-26投稿量爆炸:近3万篇论文,2万来自中国,评审系统都快崩了
人工智能·openai
AI浩2 小时前
YOLOv8-SMOT:一种高效鲁棒的实时小目标跟踪框架:基于切片辅助训练与自适应关联
人工智能·yolo·目标跟踪
兰亭妙微2 小时前
用户体验设计 | 什么是 AX?从 UX 到 AX 的演进
人工智能·交互·ux·用户体验设计公司