【报错原因】
尽管 LightGBM 模型支持直接处理分类变量,但是在后续节点中 TREESHAP 计算 LightGBM 模型的 SHAP 值时分类变量是存在一些问题的。
【解决方法】
对分类变量进行独热编码,然后再次运行模型和shap计算。
在对分类变量进行独热编码后(独热编码节点),需要对结果进行标签编码(标签编码转换节点),将 TRUE
和 FALSE
转换为 1
和 0
。此外,还需要将转换后的变量类型更改为数值型(转换变量类型节点)。在选择使用转换后的独热编码变量时,应去掉一个虚拟变量,以避免共线性问题。
# SHAP
© 版权声明
文章版权归作者所有,未经允许请勿转载。
THE END
暂无评论内容