代码之家 › 专栏 › 技术社区 › Mel

使用sklearn,我如何找到决策树的深度?

decision-tree scikit-learn python

2

Mel · 技术社区 · 7 年前

dt_clf = tree.DecisionTreeClassifier()

这个 max_depth 参数默认为 None . 根据文件,如果 没有一个 ,然后展开节点,直到所有叶都是纯叶或直到所有叶都包含小于 min_samples_split

在拟合模型后,我如何找出实际上是吗?这个 get_params() 函数没有帮助。装配后, 它仍然说 没有一个 .

我怎样才能得到你的实际电话号码 最大深度 ?

1 回复 | 直到 7 年前

1

24

Primusa 7 年前

访问 max_depth Tree 对象:

from sklearn import tree
X = [[0, 0], [1, 1]]
Y = [0, 1]
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, Y)
print(clf.tree_.max_depth)
>>> 1

help(clf.tree_)

最大深度 node_count ,以及其他较低级别的参数。