h2o (version 3.44.0.3)

h2o.get_leaderboard: Retrieve the leaderboard from the AutoML instance.

Description

Contrary to the default leaderboard attached to the automl instance, this one can return columns other than the metrics.

Usage

h2o.get_leaderboard(object, extra_columns = NULL)

Value

An H2OFrame representing the leaderboard.

Arguments

object

The object for which to return the leaderboard. Currently, only H2OAutoML instances are supported.

extra_columns

A string or a list of string specifying which optional columns should be added to the leaderboard. Defaults to None. Currently supported extensions are:

  • 'ALL': adds all columns below.

  • 'training_time_ms': column providing the training time of each model in milliseconds (doesn't include the training of cross validation models).

  • 'predict_time_per_row_ms': column providing the average prediction time by the model for a single row.

  • 'algo': column providing the algorithm name for each model.

Examples

Run this code
if (FALSE) {
library(h2o)
h2o.init()
prostate_path <- system.file("extdata", "prostate.csv", package = "h2o")
prostate <- h2o.importFile(path = prostate_path, header = TRUE)
y <- "CAPSULE"
prostate[,y] <- as.factor(prostate[,y])  #convert to factor for classification
aml <- h2o.automl(y = y, training_frame = prostate, max_runtime_secs = 30)
lb <- h2o.get_leaderboard(aml)
head(lb)
}

Run the code above in your browser using DataCamp Workspace