shap.plots.beeswarm

shap.plots.beeswarm(shap_values, max_display=10, order=shap.Explanation.abs.mean(0), clustering=None, cluster_threshold=0.5, color=None, axis_color='#333333', alpha=1, show=True, log_scale=False, color_bar=True, s=16, plot_size='auto', color_bar_label='Feature value')

创建一个 SHAP 蜂群图，当提供特征值时按特征值着色。

参数:

shap_values解释

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

max_display整数

在图中包含多少个顶级特征（默认是10，或交互图中的7）。

显示布尔

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

颜色条布尔

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

s浮动

标记的大小。更多信息请参见 matplotlib.pyplot.scatter 中的 s。

图表大小“auto”（默认），float，（float，float），或 None

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

示例

参见 beeswarm 图示例。