shap.plots.bar

shap.plots.bar(shap_values, max_display=10, order=shap.Explanation.abs, clustering=None, clustering_cutoff=0.5, show_data='auto', ax=None, show=True)

创建一组 SHAP 值的条形图。

参数:

shap_valuesshap.Explanation 或 shap.Cohorts 或 shap.Explanation 对象的字典

传递一个多行的 Explanation 对象会创建一个全局特征重要性图。

传递解释的单行（即 shap_values[0]）会创建一个局部特征重要性图。

传递一个由Explanation对象组成的字典将创建一个多条形图，每个条形类型对应于由解释对象表示的每个队列。

max_display整数

在条形图中包含多少个顶级特征（默认是10）。

顺序OpChain 或 numpy.ndarray

一个函数，给定一个SHAP值矩阵和一个轴，返回一个排序顺序，或者给定一个 numpy.ndarray 直接返回样本排序。

默认情况下，取绝对值。

聚类: np.ndarray 或 None

分区树，由 shap.utils.hclust 返回

clustering_cutoff: float

控制显示的聚类结构的多少。

show_data: bool 或 str

控制数据值是否作为 y 轴刻度标签的一部分显示。如果设置为 “auto”，我们仅在没有变换时显示数据。

ax: matplotlib Axes

要在其上绘制图形的 Axes 对象，否则使用当前的 Axes。

显示布尔

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

返回:

ax: matplotlib Axes

返回带有绘图的 Axes 对象。仅在 show=False 时返回。

示例

参见 条形图示例。