shap.plots.violin

shap.plots.violin(shap_values, features=None, feature_names=None, max_display=None, plot_type='violin', color=None, axis_color='#333333', title=None, alpha=1, show=True, sort=True, color_bar=True, plot_size='auto', layered_violin_max_num_bins=20, class_names=None, class_inds=None, color_bar_label='Feature value', cmap=<matplotlib.colors.LinearSegmentedColormap object>, use_log_scale=False)[源代码]

创建一个 SHAP 小提琴图,当提供特征值时按特征值着色。

参数:
shap_values解释,或 numpy.array

对于单一输出的解释,这是一个 SHAP 值的矩阵(# 样本 x # 特征)。

功能numpy.array 或 pandas.DataFrame 或 list

特征值矩阵(#样本 x #特征)或 feature_names 列表作为简写。

特征名称列表

功能名称(长度:# 功能)。

max_display整数

在图中包含多少个顶级特征(默认是20)。

plot_type“violin”,或“layered_violin”。

要生成哪种类型的摘要图。“layered_violin”图显示每个变量的SHAP值分布。“violin”图与此相同,只是异常值绘制为散点。

颜色条布尔

是否绘制颜色条(图例)。

显示布尔

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

图表大小“auto”(默认),float,(float,float),或 None

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

示例

参见 小提琴图示例