各位大佬,想请教~~
在训练回归tree的时候,是如何抉择初始化分割节点(哪个节点,具体value)
看书的时候说是利用贪心算法。
我的理解是 遍历所有的特征作为某个节点,然后分割value不知道该怎么选?
应该是选取loss最小的那个特征作为分割节点
不知道理解对否,谢谢!
2个回答
用左子节点的均值作为左子节点的预测值,用右子节点的均值作为右子节点的预测值,对比真实值,可以得到MSE或者MAE
遍历所有特征的所有分割点,找到MSE或者MAE最小的那个分割点
各位大佬,想请教~~
在训练回归tree的时候,是如何抉择初始化分割节点(哪个节点,具体value)
看书的时候说是利用贪心算法。
我的理解是 遍历所有的特征作为某个节点,然后分割value不知道该怎么选?
应该是选取loss最小的那个特征作为分割节点
不知道理解对否,谢谢!
用左子节点的均值作为左子节点的预测值,用右子节点的均值作为右子节点的预测值,对比真实值,可以得到MSE或者MAE
遍历所有特征的所有分割点,找到MSE或者MAE最小的那个分割点