LightGBM plot_tree() 叶子节点编号

3

LightGBM的plot_tree方法中的数字代表什么?举个例子,我使用了Pima Indian Diabettes数据集,然后使用plot_tree方法得到如下结果:

enter image description here

叶节点上的数字代表什么意思?

2个回答

0

不,我指的不是这些叶子节点的编号。问题是关于叶子节点后面的浮点数值。以上图为例,叶子节点0的值为-0.668。-0.668是什么意思? - David293836
每个决策的值应该是“信息增益”。根据LightGBM的设置,它也可以是熵、基尼不纯度等。 - Koorosh Aslansefat
谢谢!我会查一下的。 - David293836
1
请查看这里这里。请注意,叶子节点上的那些值不是像Koorosh Aslansefat所建议的“信息增益”。 - Lei

0
根据API描述,这是预测值
实际上,它是叶子节点上的残差
每个残差都被组合起来生成最终估计值。

顺便说一下,
有很多关于梯度提升决策树算法的文章,
但其中最简单的解释在这里


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接