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

Question

我使用 sklearn 制作了一个决策树，在这里，在 SciKit 学习 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) 但仅此而已。

Answer 1

pclass节点的基尼指数=左节点的基尼指数*（左节点的样本数/左节点的样本数+右节点的样本数）+右节点的基尼指数*（左节点的样本数）节点/左节点的样本数+右节点的样本数）所以这里将是Gini index of pclass = 0 + .408 *(7/10) = 0.2856