决策树 - 如何找到树中的叶子数？

Question

训练一个随机森林后，我想知道森林中每棵树的叶子数量是多少。

例如，如果我训练了一个有 10 棵树的森林，我想得到一个有 10 个条目的向量，其中第 i 个条目是第 i 个树中的叶子数。

谢谢！

Answer 1

您应该查看集成树的export_graphviz 方法的实现，以了解如何解压树数据结构。

documentation 中提供了export_graphviz 方法的示例用法。