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 对象字典将创建一个多条形图，其中每种条形类型代表由解释对象表示的 cohort。

max_displayint

要在条形图中包含多少个最重要的特征（默认为 10）。

orderOpChain 或 numpy.ndarray

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

默认情况下，取绝对值。

clustering: np.ndarray 或 None

一个分区树，由 shap.utils.hclust() 返回

clustering_cutoff: float

控制显示多少聚类结构。

show_data: bool 或 str

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

ax: matplotlib Axes

Axes 对象，用于在上面绘制图，否则使用当前的 Axes。

showbool

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

返回:

ax: matplotlib Axes

返回 Axes 对象，绘图绘制在其上。仅当 show=False 时返回。

示例

请参阅 条形图示例。