python - 如何找到决策树中每个叶子或节点的索引？

Question

主要问题是找出每个样本被分类到哪个叶节点。有成千上万的关于使用的帖子tree.apply。我很清楚这个函数，它返回叶节点的索引。

现在，我想在图的节点中添加叶子索引（我使用 Graphviz 生成）。

绘制用于索引的枚举技术将不起作用。我正在开发的决策树很大。因此，我需要能够在图中打印叶子索引。

我愿意接受的另一个选择是生成一个包含决策树叶节点的所有叶索引（以相同顺序）的数组。关于如何做到这一点的任何提示？

标签： pythonscikit-learndecision-tree

Answer 1

node_ids命令有一个参数export_graphviz。当此参数设置为True时，索引将添加到决策树的标签上。