shap.plots.waterfall

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

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

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

当模型中的特征数量超过 max_display 参数时，特征按其 SHAP 值的幅度排序，幅度最小的特征组合在绘图底部。

参数:

shap_valuesExplanation

包含要绘制的特征值和 SHAP 值的一维 Explanation 对象。

max_displayint

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

showbool

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

示例

请参阅 瀑布图示例。