在Matlab中的回归树

Question

3

我正在尝试在Matlab中实现一个梯度树提升算法，但我需要控制回归树中终端节点（叶节点）的数量。使用统计工具箱中的classregtree有办法做到吗？

谢谢。

- ACBM

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Sincole Brans · Answer 1

你可以尝试对树进行修剪。Matlab有两种修剪方式，按层级和按节点。我更喜欢按层级修剪，这样你可以指定层数，它会为你修剪。我修剪树的原因是为了避免过度拟合，如果你有一个大树的话就会发生。修剪的简单代码如下：

tprune = prune(tree,'level',p) ;

其中p为层数。您可以构建一个for循环来遍历'p'，以便在满足规格之前一直进行剪枝。