説明可能なAI⑥(具体的な貢献度を可視化)

waterfall_plot

waterfall_plotは、force_plotと表現方法が違うだけで同じように具体的な貢献度を可視化することができます。

[Google Colaboratory]

1
2
3
4
5
row_index = X_test.index.get_loc(253)
shap.plots._waterfall.waterfall_legacy(
expected_value=explainer.expected_value[0],
shap_values=shap_values[row_index,:],
features=X_train.iloc[row_index,:])

[実行結果]

force_plotでのbase_valueは、上図でE[f(X)]=22.745と表示されています。

E[f(X)]を基点として下から上へSHAP値を足し引きして、最終的にf(x)=28.55になることが確認できます。

force_plotwaterfall_plotは、表現方法が異なるだけなので好きな方を利用するようにしましょう。