shap.plots.beeswarm

shap.plots.beeswarm(shap_values: Explanation, max_display: int | None = 10, order=shap.Explanation.abs.mean(0), clustering=None, cluster_threshold=0.5, color=None, axis_color='#333333', alpha: float = 1.0, ax: Axes | None = None, show: bool = True, log_scale: bool = False, color_bar: bool = True, s: float = 16, plot_size: Literal['auto'] | float | tuple[float, float] | None = 'auto', color_bar_label: str = 'Feature value', group_remaining_features: bool = True)

创建一个 SHAP 蜜蜂群图，当提供特征值时，会根据特征值着色。

参数:

shap_valuesExplanation

这是一个 Explanation 对象，包含 SHAP 值矩阵（# 样本 x # 特征）。

max_displayint

要在图中包含多少个最重要的特征（默认为 10 个，或交互图为 7 个）。

ax: matplotlib Axes

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

showbool

是否在返回之前调用 matplotlib.pyplot.show()。将其设置为 False 允许在创建绘图后进一步自定义绘图，并通过 matplotlib.pyplot.gca() 返回当前轴。

color_barbool

是否绘制颜色条（图例）。

sfloat

标记的大小。有关更多信息，请参阅 s 在 matplotlib.pyplot.scatter() 中。

plot_size“auto”（默认），float，(float, float), 或 None

绘图的大小。默认情况下，大小会根据正在显示的特征数量自动缩放。传递单个浮点数将使每行的高度为这么多英寸。传递一对浮点数将按这么多英寸缩放绘图。如果传递 None，则当前图形的大小将保持不变。如果 ax 不是 None，则传递 plot_size 将引发 ValueError。

group_remaining_features: bool

如果特征多于 max_display，则绘制一行，表示所有剩余特征的 SHAP 值之和。默认为 True。

返回:

ax: matplotlib Axes

返回绘制了绘图的 Axes 对象。仅当 show=False 时返回。

示例

请参阅 蜜蜂群图示例。