shap.plots.image

shap.plots.image(shap_values: ~shap._explanation.Explanation, pixel_values: ~numpy.ndarray | None = None, labels: list | None = None, true_labels: list | None = None, width: int | None = 20, aspect: float | None = 0.2, hspace: float | None = 0.2, labelpad: float | None = None, cmap: str | None = <matplotlib.colors.LinearSegmentedColormap object>, show: bool | None = True)[源代码]

绘制图像输入的 SHAP 值。

参数:
shap_values[numpy.array]

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

像素值numpy.array

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

标签list 或 np.ndarray

模型输出的每个名称的列表或 ``np.ndarray``(# 样本 x top_k 类别),这些输出正在被解释。

true_labels: list

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

宽度浮动

生成的 matplotlib 图形的宽度。

labelpad浮动

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

显示布尔

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

示例

参见 图像绘制示例