h2o.varimp-H2OFrame-method function

Retrieve the variable importance.

Retrieve the variable importance.

## S4 method for signature 'H2OFrame' h2o.varimp(object, num_of_features = NULL)

Arguments

  • object: A leaderboard frame.
  • num_of_features: Integer specifying the number of features returned based on the maximum importance across the models. Use NULL for unlimited. Defaults to NULL.

Examples

## Not run: library(h2o) h2o.init() f <- "https://s3.amazonaws.com/h2o-public-test-data/smalldata/prostate/prostate_complete.csv.zip" pros <- h2o.importFile(f) response <- "GLEASON" predictors <- c("ID", "AGE", "CAPSULE", "DCAPS", "PSA", "VOL", "DPROS") aml <- h2o.automl(x = predictors, y = response, training_frame = pros, max_runtime_secs = 60) h2o.varimp(aml@leaderboard[1:5,]) ## End(Not run)
  • Maintainer: Tomas Fryda
  • License: Apache License (== 2.0)
  • Last published: 2024-01-11