scikit-learn 中保存模型是机器学习工程化落地的关键环节，其核心目标是将已训练完成的模型（包括参数、超参及内部状态）序列化为文件，以便后续加载复用、部署或跨环境迁移。官方推荐且最常用的方式是借助 joblib 库，因其对 numpy 数组高度优化，在处理如随机森林、svm、逻辑回归等含大量数值参数的模型时性能远优于原生 pickle。

---

✅ 推荐方法：使用 joblib 保存与加载

▶ 步骤一：安装依赖（若未安装）

pip install scikit-learn joblib

注：joblib 通常随 scikit-learn 自动安装，无需额外操作；若报错 ModuleNotFoundError，单独执行上述命令即可。

▶ 步骤二：训练模型

以鸢尾花数据集为例：

from sklearn.datasets import load_iris
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import train_test_split
<h1>1. 加载数据</h1><p>iris = load_iris()
X, y = iris.data, iris.target</p><h1>2. 划分训练/测试集</h1><p>X_train, X_test, y_train, y_test = train_test_split(
X, y, test_size=0.2, random_state=42
)</p><h1>3. 训练模型</h1><p>model = RandomForestClassifier(n_estimators=100, max_depth=3, random_state=42)
model.fit(X_train, y_train)</p>

▶ 步骤三：保存模型（持久化）

import joblib<h1>将模型保存为 .joblib 文件（推荐扩展名）</h1><p>joblib.dump(model, 'iris_rf_model.joblib')</p><h1>也可保存为 .pkl（兼容旧习惯，但非最优）</h1><h1>joblib.dump(model, 'iris_rf_model.pkl')</h1>

✅ 优势：joblib 对大型数组压缩效率高、读写速度快；支持自动压缩（如 .joblib.gz）、内存映射（mmap_mode）等高级特性。

▶ 步骤四：加载模型（反序列化）

# 从文件恢复模型对象（完全等价于原始训练实例）<p>loaded_model = joblib.load('iris_rf_model.joblib')</p><h1>验证可用性</h1><p>pred = loaded_model.predict([[5.1, 3.5, 1.4, 0.2]])
print("预测类别:", pred)  # 输出示例: [0]</p>

---

⚠ 其他方式（不推荐为主流方案）

• pickle 模块（Python 内置）
  虽然语法类似，但序列化 NumPy 数组较慢、体积更大，且存在安全风险（反序列化不可信文件可能导致代码执行）：

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

• sklearn.externals.joblib（已弃用）
  在 scikit-learn ≥ 0.23 版本中已被移除，请始终使用独立安装的 joblib 或 from joblib import dump, load。

---

? 最佳实践建议

项目	建议
文件命名	使用语义化名称 + 版本号，如 iris_rf_v1.2.joblib
路径管理	使用绝对路径或 pathlib.Path 确保跨平台兼容
配套保存	若含预处理器（如 StandardScaler），应一并保存整个 Pipeline
版本兼容性	joblib 保存的模型在相同 major 版本间兼容（如 joblib 1.3.x ↔ 1.3.y），跨大版本需谨慎测试

---