Learn R Programming

caretEnsemble (version 1.0.0)

varImp.caretEnsemble: Calculate the variable importance of variables in a caretEnsemble.

Description

This function wraps the varImp function in the caret package to provide a weighted estimate of the importance of variables in the ensembled models in a caretEnsemble object. Variable importance for each model is calculated and then averaged by the weight of the overall model in the ensembled object.

Usage

## S3 method for class 'caretEnsemble':
varImp(object, scale = TRUE, weight = TRUE, ...)

Arguments

object
a caretEnsemble to make predictions from.
scale
should importance values be scaled 0 to 100?
weight
should a model weighted importance be returned?
...
other arguments to be passed to varImp

Value

  • A data.frame with one row per variable and one column per model in object