我已经使用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),但仅此而已。
https://stackoverflow.com/questions/53699121
复制相似问题