Python | 网格搜索参数优化的XGBoost+SHAP可解释性分析回归预测及可视化算法

简介: 本教程将推出Python实现的XGBoost回归预测,结合网格搜索调参与SHAP可解释性分析,涵盖数据处理、模型训练、可视化及结果保存,助力科研论文提升模型可解释性,附完整代码与保姆级环境配置指南。

立个flag,这是未来一段时间打算做的Python教程,敬请关注。

1 数据及应用领域

2 算法理论基础

3 SHAP 理论基础

上述三条目录的基本原理已在前置推文中做过详细介绍,需要学习了解的请转到如下链接:

https://mp.weixin.qq.com/s/Kq6_UHmOP0Efml5ztlZOAw

✔ 程序能画非常直观的可视化

本程序SHAP带的图包括:

这些图都是发论文神器。

论文价值:可解释性直接提升一档

SCI 论文里 reviewer 最爱问:

  • “模型的物理解释是什么?”
  • “为什么这个特征如此重要?”
  • “模型是不是只是黑盒?”

你用 SHAP,一张 beeswarm plot 就能回答所有问题。

无论你是:XGBoost、CatBoost、LightGBM、Random Forest、Gradient Boosting、NGBoost、决策树,SHAP 都能解释。

4 其他图示

🎲 一、特征值相关性热图

特征值相关性热图用于展示各特征之间的相关强弱,通过颜色深浅体现正负相关关系,帮助快速识别冗余特征、强相关特征及可能影响模型稳定性的变量,为后续特征选择和建模提供参考。

🎲 二、散点密度图

散点密度图通过颜色或亮度反映点的聚集程度,用于展示大量样本的分布特征。相比普通散点图,它能更直观地呈现高密度区域、异常点及整体趋势,常用于回归分析与模型评估。以下为训练集和测试集出图效果。

🎲 三、网格搜索参数优化算法及示意图

🌟 1、网格搜索是什么?

一句话概括:

网格搜索就是把所有设定好的超参数组合排成一个“网格”,逐个尝试,通过评估结果找到表现最佳的那一组参数。

就像在一个二维或多维坐标空间里,把所有候选参数都排列出来,然后把每个点都跑一遍,最终选出模型表现最优的位置。

🌟 2、它的核心原则:全面、稳定、逐点验证

网格搜索的理念非常直观:

  • 先定义每个参数可能的取值范围
  • 再把这些取值组合成一个完整网格
  • 然后对每个组合进行模型训练与验证
  • 最后选择最优结果对应的参数

这是一种系统化、无遗漏的搜索方式。它不会遗漏,也不会偏向,它用最直接的方式告诉你:  哪个参数组合最适合你的模型。

🌟 3、为什么网格搜索常被用作调参基础流程?

网格搜索的价值主要体现在几个方面:

✔ 1. 结构清晰、可控性强

你可以完全决定参数候选集,调参过程完全透明。

✔ 2. 适用于小范围、精细化的参数探索

特别适合探索学习率、树深、正则项等关键参数的小步长变化。

✔ 3. 方便结合交叉验证

与 Cross-Validation 结合后,能够获得稳定、可靠的参数评估结果。

✔ 4. 结果可复现、可追踪

每个组合都被尝试过,调参过程完整记录,适合科研工作。

🌟 4、典型应用场景

网格搜索广泛应用于:

  • XGBoost / LightGBM / CatBoost 的关键参数精调
  • SVM、随机森林、岭回归等模型的标准调参
  • 小规模搜索空间的系统验证
  • 科研论文中要求严谨、可复现的实验设计

在你的任务里,网格搜索非常适合用于关键参数的局部精调,确保模型在最佳点附近充分探索。

🌟 5. 程序能画非常直观的可视化

该图展示 GridSearchCV 调参过程中各超参数与 RMSE 的相关性重要性,其中 learning_rate、reg_alpha 和 n_estimators 影响最明显,可用于识别关键参数并指导后续调参方向。

5 代码包含具体内容一览

我的代码程序中将参数最优值输出到当前目录的best_params.txt文本中,

并将训练集和测试集的精度评估指标保存到 metrics. Mat 矩阵中。共两行,第一行代表训练集的,第二行代表测试集的;共 7 个精度评估指标,分别代表 R, R2, ME, MAE, MAPE, RMSE 以及样本数量。

保存的regression_result.mat数据中分别保存了名字为Y_train、y_pred_train、y_test、y_pred_test的矩阵向量。

同样的针对大家各自的数据训练出的模型结构也保存在model.json中,方便再一次调用。

调用的程序我在程序中注释了,如下

# 加载模型
# model.load_model("model.json")

主程序如下,其中从1-10,每一步都有详细的注释,要获取完整程序,请转下文代码获取


# =========================================================
# 主程序
# =========================================================
def main():
   print("=== 1. 读取数据 ===")
   data = pd.read_excel("data.xlsx")
   X = data.iloc[:, :10].values
   y = data.iloc[:, 10].values
   feature_names = list(data.columns[:10])

   print("=== 2. 划分训练与测试 ===")
   X_train, X_test, y_train, y_test = train_test_split(
       X, y, test_size=0.2, random_state=42
   )

   print("=== 3. 归一化 ===")
   scaler_X = MinMaxScaler()
   scaler_y = MinMaxScaler()

   X_train_norm = scaler_X.fit_transform(X_train)
   X_test_norm = scaler_X.transform(X_test)
   y_train_norm = scaler_y.fit_transform(y_train.reshape(-1, 1)).ravel()

   print("=== 4. 模型训练 ===")
   model = train_model(X_train_norm, y_train_norm)

   print("=== 5. 预测(反归一化到原始尺度) ===")
   y_pred_train_norm = model.predict(X_train_norm)
   y_pred_test_norm = model.predict(X_test_norm)

   y_pred_train = scaler_y.inverse_transform(
       y_pred_train_norm.reshape(-1, 1)
   ).ravel()
   y_pred_test = scaler_y.inverse_transform(
       y_pred_test_norm.reshape(-1, 1)
   ).ravel()

   print("=== 6. 模型评估 ===")
   metrics_train = evaluate_model(y_train, y_pred_train)
   metrics_test = evaluate_model(y_test, y_pred_test)

   print("\n训练集评估指标:")
   for k, v in metrics_train.items():
       print(f"  {k}: {v:.4f}" if isinstance(v, float) else f"  {k}: {v}")

   print("\n测试集评估指标:")
   for k, v in metrics_test.items():
       print(f"  {k}: {v:.4f}" if isinstance(v, float) else f"  {k}: {v}")

   print("=== 7. 保存结果到 MAT 文件 ===")
   result_dict = {
       "y_train": y_train.astype(float),
       "y_pred_train": y_pred_train.astype(float),
       "y_test": y_test.astype(float),
       "y_pred_test": y_pred_test.astype(float),
   }
   savemat("regression_result.mat", result_dict)
   print("已保存 regression_result.mat")

   # 按指标顺序排列
   metrics_matrix = np.array([
       [metrics_train['R'],     metrics_test['R']],
       [metrics_train['R2'],    metrics_test['R2']],
       [metrics_train['ME'],    metrics_test['ME']],
       [metrics_train['MAE'],   metrics_test['MAE']],
       [metrics_train['MAPE'],  metrics_test['MAPE']],
       [metrics_train['RMSE'],  metrics_test['RMSE']],
       [metrics_train['样本数'], metrics_test['样本数']]
   ], dtype=float)
   savemat("metrics.mat", {"metrics": metrics_matrix})
   print("已保存 metrics.mat(矩阵大小 7×2)")

   print("=== 8. SHAP 分析 ===")
   X_combined = np.vstack([X_train_norm, X_test_norm])
   X_df = pd.DataFrame(X_combined, columns=feature_names)
   # shap_results = shap_analysis(model, X_combined, feature_names)
   plot_shap_dependence(model, X_combined, feature_names, X_df)

   print("=== 9. 密度散点图 ===")
   plot_density_scatter(
       y_test, y_pred_test, save_path="scatter_density_test.png"
   )
   plot_density_scatter(
       y_train, y_pred_train, save_path="scatter_density_train.png"
   )

   print("=== 10. 相关性热图 ===")
   correlation_heatmap(data, feature_names)

   print("=== 完成!===")

if __name__ == "__main__":
   main()

6 代码获取

Python | 网格搜索参数优化的XGBoost+SHAP可解释性分析回归预测及可视化算法

https://mbd.pub/o/bread/YZWZlppxZg==

新手小白/python 初学者请先根据如下链接教程配置环境,只需要根据我的教程即可,不需要安装 Python 及 pycharm 等软件。如有其他问题可加微信沟通。

Anaconda 安装教程(保姆级超详解)【附安装包+环境玩转指南】

https://mp.weixin.qq.com/s/uRI31yf-NjZTPY5rTXz4eA

目录
相关文章
|
机器学习/深度学习 算法 数据可视化
JAMA | 机器学习中的可解释性:SHAP分析图像复刻与解读
JAMA | 机器学习中的可解释性:SHAP分析图像复刻与解读
5542 1
|
7月前
|
机器学习/深度学习 传感器 算法
Python | Stacking回归和SHAP可解释性分析回归预测及可视化算法
本教程基于Python实现Stacking回归与SHAP可解释性分析,涵盖地球科学、医学、工程等多领域回归预测应用。结合CatBoost、LightGBM、XGBoost等模型,采用贝叶斯、随机与网格搜索优化参数,并通过SHAP值可视化特征贡献,提升模型性能与可解释性,适用于科研与实际项目。
965 2
|
机器学习/深度学习 数据可视化 算法
机器学习-可解释性机器学习:随机森林与fastshap的可视化模型解析
机器学习-可解释性机器学习:随机森林与fastshap的可视化模型解析
2130 1
|
定位技术
ArcGIS地形起伏度+地形粗糙度+地表切割深度+高程变异系数提取
ArcGIS地形起伏度+地形粗糙度+地表切割深度+高程变异系数提取
20497 0
|
9月前
|
机器学习/深度学习 数据可视化 算法
Python | 随机搜索参数优化的XGBoost+SHAP可解释性分析回归预测及可视化算法
本教程将带你掌握Python中XGBoost模型的随机搜索调参、SHAP可解释性分析及多种可视化技术,涵盖特征相关性热图、散点密度图、超参数优化等核心内容,助力科研论文与实际项目应用。
656 2
|
9月前
|
机器学习/深度学习 算法 数据可视化
Python | 贝叶斯搜索参数优化的XGBoost+SHAP可解释性分析回归预测及可视化算法
本教程将推出Python实现的XGBoost贝叶斯调参+SHAP可解释性分析与可视化,涵盖数据应用、算法原理及SHAP理论,助力SCI论文提升模型可解释性,附完整代码与环境配置指南。
1155 7
|
机器学习/深度学习 数据采集 算法
解码癌症预测的密码:可解释性机器学习算法SHAP揭示XGBoost模型的预测机制
解码癌症预测的密码:可解释性机器学习算法SHAP揭示XGBoost模型的预测机制
1774 0
|
编解码 C++
Google Earth Engine ——1958-2020年TerraClimate 全球陆地表面每月气候和气候水平衡的数据集
Google Earth Engine ——1958-2020年TerraClimate 全球陆地表面每月气候和气候水平衡的数据集
2204 0
Google Earth Engine ——1958-2020年TerraClimate 全球陆地表面每月气候和气候水平衡的数据集
|
8月前
|
机器学习/深度学习 数据采集 算法
Python | K折交叉验证的参数优化的GradientBoost及SHAP可解释性分析回归预测算法
本教程介绍基于Python的GradientBoost回归预测算法,结合K折交叉验证与贝叶斯/随机/网格搜索进行超参数优化,并引入SHAP实现模型可解释性分析。涵盖数据预处理、模型训练、多维度评估及可视化,适用于地球科学、医学、工程、经济等多个领域的连续变量预测任务,代码与数据齐全,适合科研与实际应用。
599 2
|
6月前
|
数据采集 数据处理 决策智能
【预测模型】GA-XGBoost回归+SHAP分析+新数据预测!Matlab代码实现
✅作者简介:热爱科研的Matlab仿真开发者,擅长数据处理、建模仿真、程序设计、完整代码获取、论文复现及科研仿真。 🍎 往期回顾关注个人主页:Matlab科研工作室 👇 关注我领取海量matlab电子书和数学建模资料 🍊个人信条:格物致知,完整Matlab代码获取及仿真咨询内容私信。 🔥 内容介绍 1 研究背景与意义 回归预测是数据驱动决策的核心技术,广泛应用于工业过程参数预测、环境质量评估、金融风险预测等领域。传统回归模型面临三大核心挑战: 模型超参数优
303 0