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_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 时返回。

示例

查看条形图示例。