如果您使用rpart
中的library(rpart)
函数拟合CART模型,则可以选择将分割条件指定为基尼系数或信息统计。
语法为:
rpart(y~x+z, data=df, parms=list(split='gini'))
事实上,'gini'是默认值,因此如果您只使用rpart
函数,它将使用基尼系数。有关更多信息,请参阅方法和包here的pdf简介。
如果我误解了你的用途,并且你只想要计算基尼系数的函数,你可以查看包ineq
,包括计算各种不等式的函数ineq()
。为你得分。文档为here。