matlab에서 포스트 프 루닝을위한 최대 의사 결정 트리를 설정하는 방법은 무엇입니까?

Question

내가 원하는 깊이의 나무를 어떻게 만들 수 있습니까? 예를 들어, 난 단지 3 깊이가있는 의사 결정 트리를 만들고 싶습니다.

load ionosphere 
treeModel = fitctree(X,Y) 
view(treeModel) 
view(treeModel,'mode','graph') 

이 코드는 7 깊이 트리를 만듭니다. 나는 동일한 데이터 세트를 사용하지만 3 또는 2 깊이가있는 트리를 만들고 싶습니다. 어떻게하면 MATLAB에서 할 수 있습니까?

Answer 1

MaxDepth 이름 - 값 쌍 인수를 사용하여 최대 깊이를 제어 할 수 있습니다.

자세한 내용은 the documentation을 참조하십시오.

treeModel = fitctree(X,Y,'MaxDepth',3); 

Answer 2

Matlab 환경을 구축 할 때 가능한 한 유연하게 작성하십시오. 또한 공식 설명서에 따라 MaxDepth 옵션은 높이 배열에 fitctree을 사용할 때만 적용됩니다.

load ionosphere 
treeModel = CreateTreeModel(X,Y,3); 
view(treeModel) 
view(treeModel,'mode','graph') 

function tm = CreateTreeModel(x,y,depth) 
    if (nargin < 3) 
     tm = fitctree(x,y); 
     return; 
    end 

    if (depth < 1) 
     depth = 1; 
    end 

    tm = fitctree(x,y,'MaxDepth',depth); 
end