Learn R Programming

h2o (version 3.40.0.4)

h2o.multinomial_auc_table: Retrieve the all AUC values in a table (One to Rest, One to One, macro and weighted average) for mutlinomial classification.

Description

Retrieves the AUC table from an H2OMultinomialMetrics. If "train", "valid", and "xval" parameters are FALSE (default), then the training AUC table is returned. If more than one parameter is set to TRUE, then a named vector of AUC tables are returned, where the names are "train", "valid" or "xval".

Usage

h2o.multinomial_auc_table(object, train = FALSE, valid = FALSE, xval = FALSE)

Arguments

object

An H2OMultinomialMetrics object.

train

Retrieve the training AUC table

valid

Retrieve the validation AUC table

xval

Retrieve the cross-validation AUC table

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

Run this code
if (FALSE) {
library(h2o)
h2o.init()

prostate_path <- system.file("extdata", "prostate.csv", package = "h2o")
prostate <- h2o.uploadFile(prostate_path)

prostate[, 2] <- as.factor(prostate[, 2])
model <- h2o.gbm(x = 3:9, y = 2, training_frame = prostate, distribution = "bernoulli")
perf <- h2o.performance(model, prostate)
h2o.multinomial_auc_table(perf)
}

Run the code above in your browser using DataLab