Skip to content

[SHAP]Support parameters for summary_plot #798

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

Merged
merged 1 commit into from
Sep 9, 2019

Conversation

weiguoz
Copy link
Collaborator

@weiguoz weiguoz commented Sep 9, 2019

This PR supports specify parameters for shap.summary_plot(..) with a prefix shap_summary.
Example

SELECT *
FROM iris.train
ANALYZE sqlflow_models.my_xgboost_model
WITH
    shap_summary.plot_type="bar",
    shap_summary.alpha=1,
    shap_summary.sort=True
    -- shap_summary.{k}={v}
USING TreeExplainer;

Copy link
Collaborator

@tonyyang-svail tonyyang-svail left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@weiguoz weiguoz merged commit ce0310a into sql-machine-learning:develop Sep 9, 2019
@weiguoz weiguoz deleted the shap_summary_params branch September 9, 2019 22:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants