Skip to content

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