shap.plots.text
- shap.plots.text(shap_values, num_starting_labels=0, grouping_threshold=0.01, separator='', xmin=None, xmax=None, cmax=None, display=True)
Plots an explanation of a string of text using coloring and interactive labels.
The output is interactive HTML and you can click on any token to toggle the display of the SHAP value assigned to that token.
- Parameters:
- shap_values[numpy.array]
List of arrays of SHAP values. Each array has the shap values for a string (#input_tokens x output_tokens).
- num_starting_labelsint
Number of tokens (sorted in descending order by corresponding SHAP values) that are uncovered in the initial view. When set to 0, all tokens are covered.
- grouping_thresholdfloat
If the component substring effects are less than a
grouping_threshold
fraction of an unlowered interaction effect, then we visualize the entire group as a single chunk. This is primarily used for explanations that were computed with fixed_context set to 1 or 0 when using theexplainers.Partition
explainer, since this causes interaction effects to be left on internal nodes rather than lowered.- separatorstring
The string separator that joins tokens grouped by interaction effects and unbroken string spans. Defaults to the empty string
""
.- xminfloat
Minimum shap value bound.
- xmaxfloat
Maximum shap value bound.
- cmaxfloat
Maximum absolute shap value for sample. Used for scaling colors for input tokens.
- display: bool
Whether to display or return html to further manipulate or embed. Default:
True
Examples
See text plot examples.