R/models.R
Retrieves the GINI coefficient from an H2OBinomialMetrics. If "train", "valid", and "xval" parameters are FALSE (default), then the training GINIvalue is returned. If more than one parameter is set to TRUE, then a named vector of GINIs are returned, where the names are "train", "valid" or "xval".
h2o.giniCoef(object, train = FALSE, valid = FALSE, xval = FALSE)
object | an H2OBinomialMetrics object. |
---|---|
train | Retrieve the training GINI Coefficcient |
valid | Retrieve the validation GINI Coefficcient |
xval | Retrieve the cross-validation GINI Coefficcient |
h2o.auc
for AUC, h2o.giniCoef
for the
GINI coefficient, and h2o.metric
for the various. See
h2o.performance
for creating H2OModelMetrics objects.
threshold metrics.
# NOT RUN { library(h2o) h2o.init() prosPath <- system.file("extdata", "prostate.csv", package="h2o") hex <- h2o.uploadFile(prosPath) hex[,2] <- as.factor(hex[,2]) model <- h2o.gbm(x = 3:9, y = 2, training_frame = hex, distribution = "bernoulli") perf <- h2o.performance(model, hex) h2o.giniCoef(perf) # }