决策树 - 如何查找树中的叶子数?

时间:2013-02-04 11:45:09

标签: scikit-learn

在训练了随机森林之后,我想知道森林中每棵树的叶子数量是多少。

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

谢谢!

1 个答案:

答案 0 :(得分:0)

您应该了解整体树的export_graphviz方法的实现,以了解如何解压缩树数据结构。

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