[英]How can I determine the number of trees in random Forest in matlab?
在Matlab中,我们使用TreeBagger()
方法训练随机森林。 该方法的参数之一是树的数量。 我正在使用随机森林进行分类。 如何确定随机森林的树木数量?
如果您一直在训练此模型,则应该知道模型中使用的树数,因为它必须设置为TreeBagger()
输入。
无论如何,对于像RFmodel
这样的学习模型,您可以使用compact(RFmodel)
确定树的数量。
这是基于Matlab文档的回归示例:
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
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.