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 the explainers.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.