小提琴图

使用 plot_violin() 函数创建 SHAP 小提琴图。这可用于可视化每个特征的 SHAP 值的分布和范围。

显示所有特征的特征影响分布的小提琴图示例

必填实参

实参

描述

shap_df

包含多个特征的 SHAP 值的 2D 数组

feature_df

包含相应特征值的 2D 数组

可选实参

实参

描述

figsize

控制图大小的元组,形式为 (宽度, 高度)。如未指定,则使用默认大小 (1400, 100)。

该函数会返回一个图表,该图表可视化每个特征的小提琴图。小提琴图按每个特征的绝对平均 SHAP 值排序,对模型预测影响最大的特征排在最上。

语言: 中文