Skip to content

Commit

Permalink
Stop warning for the tree path version of Tree explainer
Browse files Browse the repository at this point in the history
  • Loading branch information
slundberg committed Aug 4, 2020
1 parent f246fad commit be1fa94
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion shap/explainers/_tree.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,7 @@ def __init__(self, model, data = None, model_output="raw", feature_perturbation=
self.data = data
if self.data is None:
feature_perturbation = "tree_path_dependent"
warnings.warn("Setting feature_perturbation = \"tree_path_dependent\" because no background data was given.")
#warnings.warn("Setting feature_perturbation = \"tree_path_dependent\" because no background data was given.")
elif feature_perturbation == "interventional" and self.data.shape[0] > 1000:
warnings.warn("Passing "+str(self.data.shape[0]) + " background samples may lead to slow runtimes. Consider "
"using shap.sample(data, 100) to create a smaller background data set.")
Expand Down

0 comments on commit be1fa94

Please sign in to comment.