h2o.varimp-H2OFrame-method: Retrieve the variable importance.

h2o.varimp,H2OFrame-methodR Documentation

Retrieve the variable importance.

Description

Retrieve the variable importance.

Usage

## 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)

h2o documentation built on May 29, 2024, 4:26 a.m.