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
允许在创建图表后进一步自定义图表。
示例
参见 图像绘制示例。