【发布时间】:2015-02-08 10:45:16
【问题描述】:
decReg = DecisionTreeRegressor() clf = decReg.fit(X, Y)
直觉上,任何人都会期望 decReg 或 calf 应该有一个函数,该函数将返回树中增长的节点数。但是,我看不到任何这样的功能。还有什么要知道树的大小吗?
【问题讨论】:
标签: python machine-learning scikit-learn regression decision-tree
decReg = DecisionTreeRegressor() clf = decReg.fit(X, Y)
直觉上,任何人都会期望 decReg 或 calf 应该有一个函数,该函数将返回树中增长的节点数。但是,我看不到任何这样的功能。还有什么要知道树的大小吗?
【问题讨论】:
标签: python machine-learning scikit-learn regression decision-tree
# Instantiate a decision tree
clf = tree.DecisionTreeClassifier()
# Fit the decision tree
...
# Print node count
print(clf.tree_.node_count)
【讨论】:
根据documentation,有tree_ 属性,您可以遍历该树以查找任何感兴趣的属性。特别是,children_right 和 children_left 属性似乎很有用。
【讨论】:
tree_ 上调用dir 会有所帮助。