[英]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.