机器学习基础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%

同时也会生成模型文件

相关推荐
2zcode20 分钟前
基于Matlab的深度学习智能行人检测与统计系统
人工智能·深度学习·目标跟踪
weixin_4640780740 分钟前
机器学习sklearn:过滤
人工智能·机器学习·sklearn
weixin_4640780743 分钟前
机器学习sklearn:降维
人工智能·机器学习·sklearn
数据与人工智能律师1 小时前
智能合约漏洞导致的损失,法律责任应如何分配
大数据·网络·人工智能·算法·区块链
张艾拉 Fun AI Everyday1 小时前
小宿科技:AI Agent 的卖铲人
人工智能·aigc·创业创新·ai-native
zhongqu_3dnest1 小时前
三维火灾调查重建:科技赋能,探寻真相
人工智能
飞哥数智坊1 小时前
AI编程实战:写作助手进化,Trae+Kimi-K2两小时搞定“带样式复制”
人工智能·trae
木枷2 小时前
c2rust使用
人工智能·物联网·edge
Nayuta2 小时前
【论文导读】OS-Genesis 基于自动探索构建 GUI 数据
人工智能·机器学习
说私域2 小时前
基于开源AI智能客服、AI智能名片与S2B2C商城小程序的餐饮行业私域流量运营策略研究
人工智能·小程序·流量运营