xgboost预测概率的贡献

2024-05-16 14:12:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用xgboost的特性pred_contribs,以便为模型的每个示例获得某种可解释性(shapley值)。在

booster.predict(test, pred_contribs=True)

它返回形状(采样数)x(特征数)的贡献向量。捐款总额等于边际得分。在

但是,我想用概率来代替边际得分,为了简单起见,我想转换概率中的贡献(用近似值)。在

有办法吗?在

代码示例:

^{pr2}$

Tags: 模型testtrue示例特性贡献概率predict