shap.plots.image

shap.plots.image(shap_values: Explanation | np.ndarray | list[np.ndarray], pixel_values: np.ndarray | None = None, labels: list[str] | list[list[str]] | np.ndarray | None = None, true_labels: list | None = None, width: int | None = 20, aspect: float | None = 0.2, hspace: float | Literal['auto'] | None = 0.2, labelpad: float | None = None, cmap: str | Colormap | None = <matplotlib.colors.LinearSegmentedColormap object>, vmax: float | None = None, show: bool | None = True)

绘制图像输入的 SHAP 值。

参数:

shap_values[numpy.array]

SHAP 值数组的列表。每个数组的形状为（# 样本数 x 宽度 x 高度 x 通道数），列表的长度等于正在解释的模型输出数量。

pixel_valuesnumpy.array

每个图像的像素值矩阵（# 样本数 x 宽度 x 高度 x 通道数）。它的形状应与 shap_values 数组列表中的每个数组相同。

labelslist 或 np.ndarray

列表或 np.ndarray（# 样本数 x top_k 类别数），包含正在解释的每个模型输出的名称。

true_labels: list

要绘制的真实图像标签列表。

widthfloat

生成的 matplotlib 绘图的宽度。

labelpadfloat

在模型输出标签周围使用的填充量。

cmap: str 或 matplotlib.colors.Colormap

绘制 SHAP 值时使用的颜色映射。

vmax: Optional float

设置 SHAP 值的颜色映射范围，从 -vmax 到 +vmax。

showbool

在返回前是否调用 matplotlib.pyplot.show()。将此设置为 False 允许在创建图表后对其进行进一步的自定义。

示例

参见图像绘制示例。