matlab - matlab中的回归树

Question

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

谢谢你。

Answer 1

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

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

其中 p 是层数。您可以为 'p' 构建一个 for 循环，以便它会不断修剪，直到满足您的规范。