oegedijk / explainerdashboard

Quickly build Explainable AI dashboards that show the inner workings of so-called "blackbox" machine learning models.
http://explainerdashboard.readthedocs.io
MIT License
2.3k stars 331 forks source link

add gputree support #291

Open oegedijk opened 9 months ago

oegedijk commented 9 months ago

adds support fot shap.explainerGPUTree bu passing shap='gputree' to the explainer.

See https://shap.readthedocs.io/en/latest/example_notebooks/api_examples/explainers/GPUTree.html#