healthcareai (version 2.3.0)

get_variable_importance: Get variable importances

Description

Get variable importances

Usage

get_variable_importance(models, remove_zeros = TRUE, top_n)

Arguments

models

model_list object

remove_zeros

Remove features with zero variable importance? Default is TRUE

top_n

Integer: How many variables to return? The top_n most important variables be returned. If missing (default), all variables are returned

Value

Data frame of variables and their importance for predictive power

Details

Some algorithms provide variable importance, others don't. The best-performing model that offers variable importance will be used.

See Also

plot.variable_importance

Examples

# NOT RUN {
m <- machine_learn(mtcars, outcome = mpg, models = "rf", tune = FALSE)
(vi <- get_variable_importance(m))
plot(vi)
# }