Importance Plot for the caret::train objects — importance.train • tornado

Importance Plot for the caret::train objects

Usage

# S3 method for train
importance(model_final, ...)

Arguments

model_final: a model object
...: arguments passed to other methods

Value

an object of type importance_plot

type: the type of importance plot
data: the importance data required for the plot

See also

Examples

if (requireNamespace("caret", quietly = TRUE) &
    requireNamespace("randomForest", quietly = TRUE))
{
  model_final <- caret::train(x = subset(mtcars, select = -mpg), y = mtcars$mpg, method = "rf")
  imp <- importance(model_final)
  plot(imp)
}
#> Loading required package: ggplot2
#> Loading required package: lattice