explainerdashboard icon indicating copy to clipboard operation
explainerdashboard copied to clipboard

Support for GPUTree

Open saishashank85 opened this issue 2 years ago • 5 comments

shap.explainers.GPUTree

saishashank85 avatar Aug 25 '23 09:08 saishashank85

For the time being you can manually add this with:

explainer = ClassifierExplainer(model, X, y)
explainer.shap_explainer = GPUTree(model, X)
ExplainerDashboard(explainer).run()

oegedijk avatar Dec 18 '23 19:12 oegedijk

I will see however if I can add it add as option (although do not have a cuda device readily available for testing)

oegedijk avatar Dec 18 '23 19:12 oegedijk

Made a PR here: https://github.com/oegedijk/explainerdashboard/pull/291

could you check if you can get this to work?

oegedijk avatar Dec 18 '23 20:12 oegedijk

@oegedijk I do have a cuda device, for testing. I was thinking of sending a PR that shows OS and python compatibility.

datatalking avatar Jan 01 '24 19:01 datatalking

cool, let me know if #291 works and can be merged or needs some modifications

oegedijk avatar Jan 02 '24 08:01 oegedijk