如何获取训练好的sklearn.tree.DecisionTreeClassifier中叶子和节点的数量?

问题描述 投票:0回答:1

我有一个经过训练的决策树分类器,我想知道它有多少个叶子和节点。

python scikit-learn decision-tree
1个回答
0
投票

它们存储在内部“tree_”对象中:

clf = DecisionTreeClassifier()
clf.fit(X, y)
print('Leaves:', clf.tree_.n_leaves)
print('Leaves and Nodes:', clf.tree_.node_count)
© www.soinside.com 2019 - 2024. All rights reserved.