matlab - matlab中的回归树

Question

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

谢谢。

Answer 1

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

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

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