如何在决策树中查找每个叶或节点的索引-sklearn

问题描述 投票:1回答:1

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

现在,我想在图的节点中添加叶子索引(通过使用Graphviz生成)。

[如果你们中的某些人正在考虑将用于索引的枚举技术用作注释,那么该选项将不起作用。我正在开发的决策树很大。因此,我需要能够在图表中打印叶子索引(否则,将使我感到难以手动枚举它们)。

我愿意接受的另一种选择是使用决策树的叶节点的所有叶索引(以相同顺序)生成一个数组。有关如何执行此操作的任何提示?

谢谢大家的帮助

python scikit-learn decision-tree
1个回答
0
投票

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

© www.soinside.com 2019 - 2024. All rights reserved.