首页 > 解决方案 > SHAP:如何获得分类变量的特征重要性?

问题描述

我正在尝试将 SHAP 用于 ANN 模型解释。我发现当我使用

shap.summary_plot(shap_values[0], backgournd, plot_type='bar')

对于 dummy 之后的分类变量,特征重要性类似于

CategoricalFeature_A=0.1
CategoricalFeature_B=0.3
CategoricalFeature_C=0.3

我想知道有什么方法可以直接获得特征重要性CategoricalFeature吗?

标签: pythonshap

解决方案


推荐阅读