h2o (version 3.0.0.22)

h2o.auc: Retrieve the AUC

Description

Retrieves the AUC value from an H2OBinomialMetrics.

Usage

h2o.auc(object, valid = FALSE, ...)

Arguments

object

An H2OBinomialMetrics object.

valid

Retrieve the validation AUC

...

extra arguments to be passed if `object` is of type H2OModel (e.g. train=TRUE)

See Also

h2o.giniCoef for the Gini coefficient, h2o.mse for MSE, and h2o.metric for the various threshold metrics. See h2o.performance for creating H2OModelMetrics objects.

Examples

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.auc(perf)