MLOps Academy · 课时

使用 Pickle 和 Joblib 序列化

以安全的方式保存和加载 sklearn 模型

第 1 / 4 课13 个步骤

使用 Pickle 和 Joblib 序列化 是 CoddyKit 上的免费 MLOps Academy 课时。 这是第 1 节课,共 4 节。 你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。 这是 MLOps Academy 学习路径的一部分,你的进度在网页和 CoddyKit 应用中同步。 MLOps Academy 课程共包含 4 节课。

为什么要序列化模型

训练好的模型存在于内存中,脚本结束时就会消失。序列化会将它保存到文件中,以便稍后再次加载。💾

认识 Pickle

Python 内置的 pickle 模块可以将几乎任何对象转换为字节,存储到磁盘后再按原样重建。

将模型转储到磁盘

请以二进制写入模式打开文件,然后调用 pickle.dump 将训练好的模型写入其中。

import pickle
with open('model.pkl', 'wb') as f:
    pickle.dump(model, f)

重新加载模型

要恢复模型,请以二进制读取模式打开文件,然后调用 pickle.load。您会得到同一个对象,可以直接进行预测。

with open('model.pkl', 'rb') as f:
    model = pickle.load(f)

为什么会有 Joblib

Scikit-learn 会在模型中存储大型 NumPy 数组。与普通 pickle 相比,joblib 存储这些数组的效率高得多。🚀

使用 Joblib 保存

joblib 提供了简洁的一行代码:joblib.dump 接收对象和文件名,无需手动处理文件句柄。

import joblib
joblib.dump(model, 'model.joblib')

使用 Joblib 加载

加载过程与保存过程相似。使用相同的文件名调用 joblib.load,即可取回训练好的估计器。

model = joblib.load('model.joblib')
model.predict(X_test)

Pickle 的重大警告

永远不要反序列化来自不可信来源的文件。恶意的 pickle 会在您加载它的瞬间运行任意代码。⚠️

版本必须匹配

Pickle 文件与库版本绑定。在不同的 scikit-learn 版本下加载模型,可能会失败或悄悄产生错误行为。

固定保存时的版本

请始终在文件旁记录确切的库版本。固定您的依赖项,意味着保存的模型明天仍会以相同方式加载。

import sklearn
print(sklearn.__version__)

Pickle 还是 Joblib

对于 scikit-learn 模型,请优先使用 joblib;对于小型且不含数组的 Python 对象,普通 pickle 完全够用。

快速检查

现在测试一下您对模型序列化知识的掌握情况。

回顾

现在,您可以使用 pickle 和 joblib 保存并重新加载模型;对于 sklearn 优先使用 joblib,绝不加载不可信文件,并固定版本。🎉

免费开始

用 AI 导师学习 Python — 免费

在浏览器中编写并运行真实代码,获得全天候 AI 导师的即时帮助,并在网页或应用中继续学习。

课程
30
课程
120

常见问题解答

「使用 Pickle 和 Joblib 序列化」课时是免费的吗?

是的 — 「使用 Pickle 和 Joblib 序列化」的完整文本可在网页上免费阅读。要进行交互式练习(内置代码编辑器和全天候 AI 导师)并解锁 MLOps Academy 课程的其余内容,请升级到 CoddyKit PRO。 MLOps Academy 课程共包含 4 节课。

「使用 Pickle 和 Joblib 序列化」这节课中我会学到什么?

以安全的方式保存和加载 sklearn 模型 你通过在浏览器中直接运行的动手代码来练习 MLOps Academy,全天候 AI 导师会在你学习这节课的过程中回答你的问题。

学习 MLOps Academy 需要有经验吗?

无需任何先前经验。CoddyKit 上的 MLOps Academy 课程适合初学者到高级学习者,你可以从这里开始或从头开始,按照自己的节奏学习。 这是第 1 节课,共 4 节。

「使用 Pickle 和 Joblib 序列化」课时需要多长时间?

大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动,所以你能稳步进步,并在网页和应用中从离开的地方继续。

我能在这节 MLOps Academy 课中编写并运行代码吗?

能。每节 MLOps Academy 课都包含内置代码编辑器,你可以在浏览器中直接编写并运行真实代码,并获得即时 AI 反馈 — 无需本地设置。

此课程中的所有课时

  1. 使用 Pickle 和 Joblib 序列化
  2. MLflow 模型风格
  3. 导出为 ONNX,提升可移植性
  4. 定义模型签名和模式
← 返回 MLOps Academy