Retrieves the pr_auc value from an H2OBinomialMetrics. If "train", "valid", and "xval" parameters are FALSE (default), then the training pr_auc value is returned. If more than one parameter is set to TRUE, then a named vector of pr_aucs are returned, where the names are "train", "valid" or "xval".
h2o.pr_auc(object, train = FALSE, valid = FALSE, xval = FALSE)
| object | An H2OBinomialMetrics object. |
|---|---|
| train | Retrieve the training pr_auc |
| valid | Retrieve the validation pr_auc |
| xval | Retrieve the cross-validation pr_auc |
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.
# 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.pr_auc(perf) # }