1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
% Generated by roxygen2: do not edit by hand
% Please edit documentation in R/gini.R
\name{gini}
\alias{gini}
\alias{gini.default}
\alias{gini.glm}
\alias{gini.randomForest}
\alias{gini.glmerMod}
\alias{gini.gbm}
\alias{gini.rpart}
\title{GINI Coefficient}
\usage{
gini(...)
\method{gini}{default}(actual, predicted, ...)
\method{gini}{glm}(modelObject, ...)
\method{gini}{randomForest}(modelObject, ...)
\method{gini}{glmerMod}(modelObject, ...)
\method{gini}{gbm}(modelObject, ...)
\method{gini}{rpart}(modelObject, ...)
}
\arguments{
\item{\dots}{additional parameters to be passed the the s3 methods}
\item{actual}{A vector of the labels. Can be \code{numeric, character, or factor}}
\item{predicted}{A vector of predicted values}
\item{modelObject}{the model object. Currently supported \code{glm, randomForest, glmerMod, gbm}}
}
\description{
Calculates the GINI coefficient for a binary classifcation model
}
\examples{
data(testDF)
glmModel <- glm(y ~ ., data = testDF, family="binomial")
Preds <- predict(glmModel, type = 'response')
gini(testDF$y, Preds)
# using s3 method for glm
gini(glmModel)
}
|
