如何解决检查SHAP功能重要性对于一部分功能
我正在尝试检查使用SHAP构建的模型的功能重要性。
使用以下命令可以很好地工作,但是我希望有一个仅包含部分功能的图形。这可能吗?
shap.summary_plot(shap_values,features=X_train2,feature_names=X_train2.columns,max_display=30)
我尝试在数据集上使用iloc来定义要素和要素名称的子集,例如features=X_train.iloc[;23:]
,但是它不起作用。
解决方法
您将需要同时对特征和shap_values进行子集设置,否则它们将无法对齐。值得的是,使用shap的新SHAP API值得。Expanation对象将使此操作变得更容易,因为它允许同时对要素和SHAP值进行简单的并行切片。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。