matlab에 회귀 트리

Question

matlab에 그라디언트 트리 부스팅 알고리즘을 구현하려고하지만 회귀 트리에서 터미널 노드 (리프 노드)의 수를 제어해야합니다. 통계 도구 상자의 classregtree를 사용하여 그렇게 할 수 있습니까?

감사합니다.

Answer 1

트리 제거를 시도 할 수 있습니다. Matlab은 레벨과 노드별로 두 가지 방식으로 가지 치기를합니다. 나는 당신이 레벨의 수를 지정할 수 있도록 레벨별로 선호하며, 당신을 위해 그것을자를 것입니다. 내가 나무를 가지 치기 시작한 이유는 나무가 너무 크면 나무에 과장되는 것을 피하기 위해서입니다. 자두에 대한 간단한 코드 p는 수준의 수입니다

tprune = prune(tree,'level',p) ; 

입니다. 'p'에 대한 for 루프를 구성하여 스펙이 충족 될 때까지 프 i (prune)을 유지할 수 있습니다.