DecisionTreeClassifier和DecisionTreeRegressor的tree._feature最后几次返回值-2。这是因为它们是叶节点吗?我可以假设任何-2值作为叶节点特征吗?
[通常,是的。当该节点上没有拆分时,对象变量.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
。