sklearn.tree.tree.Tree中的容量和node_counts有什么区别？

Question

我想从python中导出决策树结构。因此，我考虑保存树的每个节点。在sklearn的Tree文档中，我发现有两个属性capacity和node_counts非常相似。在我目前的使用中，它们总是相同的。但我想知道他们的主要区别是什么？

https://www.kite.com/python/docs/sklearn.tree.tree.Tree

Answer 1

从您引用的文档中：

• node_count-
• node_count - current number of nodes.
• capacity -此树在需要调整内部缓冲区大小之前可以容纳的最大节点数。

node_count始终为<=容量。

同时，树不希望将容量增加到node_count之外太多，因为未使用的内部缓冲区会消耗相当多的内存。

在编写一个树导出程序时，你应该关注node_count，而不是容量。