是否有任何函数可以计算R中的CART（决策树算法）的基尼指数？

Question

使用CART， 我想使用基尼索引从整个属性中选择主要属性。

但我找不到包含它的任何函数或包。

如果有任何函数或包计算基尼指数，请让我知道。

Answer 1

如果您使用library(rpart)中的rpart函数拟合CART模型，则可以选择将分割标准指定为基尼系数或信息统计。

的语法是：

rpart(y~x+z, data=df, parms=list(split='gini')) 

事实上，“基尼”是默认的，所以如果你只需要使用rpart功能将反正用基尼系数。欲了解更多，你可以看到方法和包here的pdf介绍。

如果我误会了你所使用的本作，只是想和你说计算基尼系数的功能，你可以看包ineq，包括功能ineq()这将计算各种不平等的分数为你的。文档是here。