scikit学习功能中的DecisionTree分类器返回值-2,这意味着什么

问题描述 投票:-1回答:1

DecisionTreeClassifier和DecisionTreeRegressor的tree._feature最后几次返回值-2。这是因为它们是叶节点吗?我可以假设任何-2值作为叶节点特征吗?

python-3.x machine-learning scikit-learn data-science decision-tree
1个回答
0
投票

[通常,是的。当该节点上没有拆分时,对象变量.tree_.feature返回-2,只有当该节点是叶子时(当树被修剪而没有修剪时,即fit(..., ccp_alpha=0)时),该变量才会发生。

虽然help()文档中没有100%清楚,但是可以在以下代码内找到对此的引用:https://github.com/scikit-learn/scikit-learn/blob/master/sklearn/tree/_tree.pyx

[feature值设置为静态变量TREE_UNDEFINED = -2

© www.soinside.com 2019 - 2024. All rights reserved.