-
-
Notifications
You must be signed in to change notification settings - Fork 333
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Support for GPUTree #281
Comments
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: #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 |
shap.explainers.GPUTree
The text was updated successfully, but these errors were encountered: