Control Decision Tree Depth

Question

Ryan Jones 2020-11-29

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/668108-control-decision-tree-depth

评论： 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.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

Pratyush Roy 2020-12-22

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/668108-control-decision-tree-depth#answer_581960

编辑：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 个评论
显示 -1更早的评论隐藏 -1更早的评论

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.

请先登录，再进行评论。