How to display shap waterfall plot

invalidargument
New Contributor III

Hi,

I have managed to display force plot for a single observation using the advice from this thread:

Solved: How to display SHAP plots? - Databricks - 28315

But is there anyway to display the newer "waterfall"-plot 

shap.plots.waterfall — SHAP latest documentation

When running

 

shap.plots.waterfall(shap_values[0])

 

 I get output like

 

Out[105]: <Figure size 576x468 with 3 Axes>

 

 I have tried calling:

  • plt.show()
  • display()
  • display(plt.gcf())

But same result. Any advice? Is it supposed to work?