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
,则当前图形的大小将保持不变。
示例
参见 小提琴图示例。