Evalml: Add partial dependence plot

Created on 28 Jul 2020  ·  3Comments  ·  Source: alteryx/evalml

All 3 comments

One thing that didn't occur to me before is that we can actually use shap values to do the dependency plot. https://slundberg.github.io/shap/notebooks/plots/dependence_plot.html

I suspect we still start by just adding traditional partial deps plots, but wanted to share this since it is interesting and something to keep in mind as we develop this API.

@kmax12 Thanks for sharing this! I agree that it'd make sense to start by adding traditional partial dependency plots, so I filed #1068 to track using SHAP values to create dependency plots to make sure this doesn't get lost when this issue is closed.

Discussed with @angela97lin . Takeaways:

  • For now we'll support pipelines. We won't add test coverage / docs for accepting estimators as input, unless it turns out to "just work" and only takes another couple hours to add unit test coverage
  • We'll only generate 1d plots, not 2d interaction plots
  • API:
def partial_dependence(pipeline, X, feature_name, grid_resolution=None):
    ...
def graph_partial_dependence(pipeline, X, feature_name, grid_resolution=None):
    ...
Was this page helpful?
0 / 5 - 0 ratings