是否有任何函数可以计算R中CART(决策树算法)的基尼指数?

时间:2016-11-04 12:53:33

标签: r cart gini

在使用CART时, 我想使用Gini索引从整个属性中选择主要属性。

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

如果有任何计算基尼指数的函数或包,请告诉我。

1 个答案:

答案 0 :(得分:0)

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

语法为:

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

事实上,'gini'是默认值,因此如果您只使用rpart函数,它将使用基尼系数。有关更多信息,请参阅方法和包here的pdf简介。

如果我误解了你的用途,并且你只想要计算基尼系数的函数,你可以查看包ineq,包括计算各种不等式的函数ineq()。为你得分。文档为here