我正在尝试在matlab中实现梯度树增强算法,但是我需要控制回归树中终端节点(叶节点)的数量。有没有一种方法可以使用统计工具箱中的classregtree?
谢谢。
最佳答案
您可以尝试修剪树。 Matlab按级别和按节点两种方式进行修剪。我更喜欢按级别,以便您可以指定级别数,它会为您修剪。
我修剪树的原因是为了避免过度拟合树,如果您有一棵大树,则会发生这种情况。修剪的简单代码是
tprune = prune(tree,'level',p) ;
其中p是级别数。您可以为'p'构造一个for循环,以便它会一直修剪直到满足您的要求。