如何知道使用Scikit-learn构建的树的大小(节点数)?

时间:2014-12-10 09:39:51

标签: python machine-learning scikit-learn regression decision-tree

decReg = DecisionTreeRegressor() clf = decReg.fit(X,Y)

直观地,任何人都会期望decReg或calf应该具有一个函数,该函数将返回树中生长的节点数。但是,我无法看到任何此类功能。还有什么可以知道树的大小吗?

2 个答案:

答案 0 :(得分:1)

根据documentationtree_属性,您可以遍历该树以查找任何感兴趣的属性。特别是,children_rightchildren_left属性似乎很有用。

答案 1 :(得分:0)

`clf = tree.DecisionTreeClassifier()

装入决策树后

x = clf.tree _

打印(x.node_count)`