shap.plots.waterfall

shap.plots.waterfall(shap_values, max_display=10, show=True)

绘制单个预测的解释作为瀑布图。

特征的 SHAP 值表示该特征提供的证据对模型输出的影响。瀑布图旨在直观地展示每个特征的 SHAP 值（证据）如何将模型输出从背景数据分布下的先验期望移动到所有特征证据给定后的最终模型预测。

特征按其 SHAP 值的大小排序，当模型中的特征数量超过 max_display 参数时，最小幅度的特征会分组显示在图表的底部。

参数:

shap_values解释

一个一维的 Explanation 对象，包含特征值和SHAP值以供绘图。

max_display整数

要显示的最大特征数量（默认是10）。

显示布尔

是否在返回之前调用 matplotlib.pyplot.show() 。将其设置为 False 允许在创建图表后进一步自定义，通过 plt.gca() 返回当前轴。

示例

参见 瀑布图示例。