iterative / example-repos-dev

Source code and generator scripts for example DVC projects
https://dvc.org/doc
21 stars 13 forks source link

get-started: add shap plot #136

Closed dberenbaum closed 1 year ago

dberenbaum commented 1 year ago

Started on a SHAP plot. This is a popular method for feature importance that's more robust than the ones built in to sklearn. It provides an example of a plot that needs to be saved as an image since it would be hard to support as a vega plot.

Example output:

image

One downside is that it's expensive, so it might make sense to sample the data here, but I didn't bother for now.

dberenbaum commented 1 year ago

Closing as stale