Control Decision Tree Depth

8 次查看（过去 30 天）
Ryan Jones
Ryan Jones 2020-11-29
评论： Ryan Jones 2021-1-22
MATLAB's function,
fitctree
has name-value arguement to control the maximum number of branch node splits, the minimum leaf size and the minimum parent node size.
I would like to compare two different feature matricies with the same dataset. I want to evaluate the training error and CV error for each model built using the two different feature matrices. However, for a fair compaison, I would like to compute these errors with models of the same tree depth and I can't find a way to specify the number of levels I want the trees to have, nor can I find a pruning method that prunes by tree levels and not by nodes.
Does anyone have any ideas of what I can do? Thanks.

请先登录，再回答此问题。

回答（1 个）

Pratyush Roy
Pratyush Roy 2020-12-22
编辑：Pratyush Roy 2020-12-22
There is no direct way to set the depth to which we want to grow the tree. This issue has been raised to the concerned people and they might be considered in the future releases of the MATLAB.
EDIT: I have received communication regarding a workaround. For tall arrays, one can use the 'MaxDepth' name-value property for setting the maximum depth to which we want to grow the tree.
The following link might be helpful:
https://www.mathworks.com/help/stats/fitctree.html#namevaluepairarguments
Hope this helps!
  1 个评论
Ryan Jones
Ryan Jones 2021-1-22
Thank you, that is good to know, so I don't keep trying to find methods of fixing the depth.
However, using the method using tall array results in a compact tree and you can't do basic operations such as cross-validation. Plus, I belive it takes longer to build than normal classification trees as it has to evaluate the tall arrays.

请先登录，再进行评论。

类别

在 Help Center 和 File Exchange 中查找有关 Classification Trees 的更多信息

标签

产品


版本

R2020b

Community Treasure Hunt

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

Start Hunting!

Translated by