Python中如何保存和加载训练好的模型_joblib进行模型持久化序列化

冬婷姑娘_6301

冬婷姑娘_6301

2026-05-06

310人浏览

原创

应优先用 joblib.dump() 保存纯 sklearn 模型或含大量 numpy 数组的模型,因其比 pickle 快 2–10 倍且更省内存;但跨 python 版本、sklearn 版本或含自定义对象时需改用 pickle。

python中如何保存和加载训练好的模型_joblib进行模型持久化序列化

用 joblib 保存和加载模型,比 pickle 更快、更省内存,尤其适合 sklearn 模型;但它不通用,不能跨 Python 版本或跨平台可靠读取。

什么时候该用 joblib.dump() 而不是 pickle.dump()

sklearn 官方明确推荐 joblib:它对 numpy 数组做了专门优化,序列化大型模型(比如 RandomForestClassifier、LinearRegression)时速度快 2–10 倍,文件体积也更小。但注意:joblib 不是通用序列化工具——如果你模型里混用了自定义类、lambda 函数或非标准对象,joblib 可能失败,这时得切回 pickle 或重构代码。

  • ✅ 推荐场景:纯 sklearn 模型、含大量 numpy.ndarray 的模型(如 SGDClassifier 的 coef_)
  • ❌ 避免场景:模型里用了 functools.partial、闭包、未导出的本地类
  • ⚠️ 注意:joblib.load() 必须和保存时用的 Python 版本、sklearn 版本尽量一致,否则可能报 ModuleNotFoundError 或反序列化出错

joblib.dump() 的关键参数怎么设才安全

最常漏掉的是 compress 和 protocol 参数。默认 compress=0 不压缩,大模型会生成巨量文件;而 protocol 默认用当前 Python 的最高协议(如 Python 3.9 用 protocol 5),但换到旧环境就打不开。

python-parallelization
python-parallelization

将顺序 Python 代码转换为并行/并发实现,用于在需要并行化 Python 代码、提升性能时使用。

下载
  • compress=3 是实用平衡点:压缩率够高,CPU 开销低;值可选 0(无压缩)、1–9(数字越大越慢但越小)
  • protocol=4 兼容性最好:Python 3.4+ 都支持,避免部署到旧服务器时报错
  • 务必显式指定 filename 后缀为 .joblib(如 "model.joblib"),别用 .pkl,否则容易混淆用途
joblib.dump(model, "model.joblib", compress=3, protocol=4)

加载模型时 joblib.load() 报错常见原因

错误信息像 ValueError: Unsupported pickle protocol 或 ModuleNotFoundError: No module named 'sklearn.ensemble._forest',基本都指向环境不一致。

  • 检查 Python 小版本是否一致(如 3.8.10 vs 3.8.12 通常 OK,但 3.8 vs 3.9 就不行)
  • sklearn 版本差异超过一个 minor 版(如 1.1.x → 1.3.x)可能导致内部模块路径变化,直接 load 失败
  • 如果模型保存在 Windows 上,加载到 Linux 时路径分隔符一般不影响,但要注意模型里是否硬编码了 Windows 路径(比如预处理 pipeline 里用了 open(r"C:\data\map.json"))
  • 临时解法:用 python -c "import joblib; print(joblib.__version__)" 对齐两边版本

生产环境部署前必须验证的两件事

只保存模型不测试加载,等于没存。尤其当模型要上 Docker 或 serverless 环境时,容易忽略依赖链。

  • 在目标环境里执行一次完整流程:joblib.load() → model.predict(X_sample) → 核对输出 shape 和 dtype 是否符合预期
  • 确认所有依赖包版本写死在 requirements.txt 中,特别是 scikit-learn、numpy、joblib 三个包,不要用 ~> 或 ^

joblib 的坑不在语法,而在环境漂移——同一段 dump/load 代码,在你本地跑通,不代表在 CI/CD 流水线或 Kubernetes Pod 里也能跑通。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1731

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4424

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1709

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

25917

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3147

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3167

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1183

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2463

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习