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 decending order by corresponding SHAP values) that are uncovered in the initial view. When set to 0 all tokens 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 Partition explainer, since this causes interaction effects to be left on internal nodes rather than lowered.

separatorstring

The string seperator that joins tokens grouped by interation effects and unbroken string spans.

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