问如何获得决策树中的所有基尼指数？
EN

Stack Overflow用户

提问于 2018-12-10 11:29:55

回答 2查看 4.3K关注 0票数 6

我已经使用sklearn创建了一个决策树，在这里，位于SciKit learn DL包下，即。sklearn.tree.DecisionTreeClassifier().fit(x,y)。

如何在每个步骤中获得所有可能节点的基尼指数？graphviz只给出基尼指数最低的节点的基尼指数，即用于拆分的节点。

例如，下面的图像(来自graphviz)告诉我Pclass_lowVMid右指数的基尼指数是0.408，但不是Pclass_lower或Sex_male在该步骤的基尼指数。我只知道Pclass_lower和Sex_male的基尼指数必须大于(0.408*0.7 + 0)，但仅此而已。

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/53699121

复制

相似问题

问如何获得决策树中的所有基尼指数？EN