我正在对决策树进行一些特征归纳,并且想知道树的大小(以节点数表示)。我该如何在 python 中做到这一点?
使用 sklearn 网站上的股票示例,
x = [[0,0],[0,1]]
y = [0,1]
from sklearn.ensemble import RandomForestClassifier
clf = RandomForestClassifer(n_estimators = 10)
clf = clf.fit(x,y)
我可以通过 clf[1]、clf[...] 等方式获取各个树,但是如何根据总节点数确定每棵树的大小?
最佳答案
sklearn.tree._tree.Tree
对象具有 node_count
属性:
from sklearn import tree
X = [[0, 0], [1, 1]]
Y = [0, 1]
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, Y)
treeObj = clf.tree_
print treeObj.node_count
关于python - 如何找出 sklearn 决策树的大小?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/26602369/