get_variable_importance: Get variable importances

Description Usage Arguments Details Value See Also Examples

View source: R/variable_importance.R

Description

Get variable importances

Usage

1
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

Details

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

Value

Data frame of variables and their importance for predictive power

See Also

plot.variable_importance

Examples

1
2
3
m <- machine_learn(mtcars, outcome = mpg, models = "rf", tune = FALSE)
(vi <- get_variable_importance(m))
plot(vi)

healthcareai documentation built on Sept. 2, 2018, 1:03 a.m.