Number of leaf nodes used by regression tree

조회 수: 4 (최근 30일)
Maaz Ahmad
Maaz Ahmad 2021년 4월 3일
답변: Aditya Patil 2021년 4월 7일
Is there any way to know the exact number of leaf nodes in a trained regression tree (fitrtree) or each tree of a regression tree ensemble (fitrensemble)? Understand that hyperparameter optimization allows one to view the MaxNumSplits which give an idea of maximum branch nodes, and hence max number of leaf nodes (which should be MaxNumSplits+1, considering root node being a leaf node, for 0 split). However, the optimal number of branchings used in the trained model may be less than MaxNumSplits. How can we obtain this information from a trained regression tree model?

답변 (1개)

Aditya Patil
Aditya Patil 2021년 4월 7일
The IsBranchNode Property of RegressionTree is false for leaf nodes. So you can use it as follows,
X = rand(100, 4);
y = rand(100, 1);
mdl = fitrtree(X,y);
leaves = sum(~mdl.IsBranchNode);

카테고리

Help CenterFile Exchange에서 Regression Tree Ensembles에 대해 자세히 알아보기

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by