开始使用免费开始使用

用 SHAP 找出关键的医疗费用预测因子

SHAP 值可以为机器学习模型的预测提供有见地的解释。现在,您将使用 SHAP 来解析 RandomForestRegressor model 中各个特征对保险费用预测的影响。

已为您预先加载了包含预测特征的 X、保险费用的 y,以及 RandomForest 回归器 model

请注意,代码运行可能需要一些时间。

本练习是课程的一部分

Python 可解释性 AI

查看课程

练习说明

  • 为 RandomForest model 初始化名为 explainer 的 SHAP 树解释器。
  • 计算数据集的 shap_values
  • 计算 SHAP 值的平均绝对值,以识别影响最大的特征。

交互式实操练习

通过完成这段示例代码来试试这个练习。

import shap

# Create a SHAP Tree Explainer
explainer = ____

# Calculate SHAP values
shap_values = ____

# Calculate mean absolute SHAP values
mean_abs_shap = ____

plt.bar(X.columns, mean_abs_shap)
plt.title('Mean Absolute SHAP Values for RandomForest')
plt.xticks(rotation=45)
plt.show()
编辑并运行代码