如何在matlab中确定随机森林中的树的数量？

Question

在Matlab中，我们使用TreeBagger()方法对随机森林进行训练。此方法的参数之一是树的数量。我使用随机森林作为分类方法。如何确定随机森林的树木数量？

Answer 1

如果您一直在训练这个模型，您应该知道模型中使用的树的数量，因为它必须设置为TreeBagger()的输入。

无论如何，对于像RFmodel这样的学习模型，您可以使用compact(RFmodel)来确定树的数量。

这是基于Matlab documentation的回归示例：

load imports-85;
Y = X(:,1);
X = X(:,2:end);
isCat = [zeros(15,1);ones(size(X,2)-15,1)]; % Categorical variable flag
rng(1945,'twister')
UnknownNumberofTrees=100;
RFmodel = TreeBagger(UnknownNumberofTrees,X,Y,'Method','R','OOBPred','On',...
        'Cat',find(isCat == 1),'MinLeaf',5);

RFmodelObject = compact(RFmodel);
RFmodelObject.NTrees

%ans =

%   100