Error in ifelse(decision_type %in% c(“>=”, “>”), ret.second(split_index), : Unknown decision_type

【报错原因】

尽管 LightGBM 模型支持直接处理分类变量,但是在后续节点中 TREESHAP 计算 LightGBM 模型的 SHAP 值时分类变量是存在一些问题的。

【解决方法】

对分类变量进行独热编码,然后再次运行模型和shap计算。

20250414142755250-image

在对分类变量进行独热编码后(独热编码节点),需要对结果进行标签编码(标签编码转换节点),将 TRUEFALSE 转换为 10。此外,还需要将转换后的变量类型更改为数值型(转换变量类型节点)。在选择使用转换后的独热编码变量时,应去掉一个虚拟变量,以避免共线性问题。

# SHAP 

© 版权声明
THE END
喜欢就支持一下吧
点赞5 分享
评论 抢沙发

请登录后发表评论

    暂无评论内容