获取决策树中特定叶节点的数量

Question

我有一个决策树，我想获取单个类的叶节点数。使用这个示例，我们得到树：

我知道可以通过

获取总节点数

clf.tree_.node_count

以及

的总叶节点数

clf.tree_.n_leaves

但是是否有可能获得紫色叶子节点的数量，即“class = virginica”的叶子节点？

Answer 1

您可以在树上运行简单的 DFS 或 BFS 算法。将计数设置为 0。对于遇到的每个节点，如果没有右子节点、左子节点且类别为

virginica

，则增加计数。