explainerdashboard
explainerdashboard copied to clipboard
Support for GPUTree
shap.explainers.GPUTree
For the time being you can manually add this with:
explainer = ClassifierExplainer(model, X, y)
explainer.shap_explainer = GPUTree(model, X)
ExplainerDashboard(explainer).run()
I will see however if I can add it add as option (although do not have a cuda device readily available for testing)
Made a PR here: https://github.com/oegedijk/explainerdashboard/pull/291
could you check if you can get this to work?
@oegedijk I do have a cuda device, for testing. I was thinking of sending a PR that shows OS and python compatibility.
cool, let me know if #291 works and can be merged or needs some modifications