Contrary to the default leaderboard attached to the automl instance, this one can return columns other than the metrics.
h2o.get_leaderboard(object, extra_columns = NULL)
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:
|
An H2OFrame representing the leaderboard.
# NOT RUN { 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) # }