healthcareai (version 2.3.0)

plot.variable_importance: Plot variable importance

Description

Plot variable importance

Usage

# S3 method for variable_importance
plot(x, title = "model", max_char = 40,
  caption = NULL, font_size = 11, point_size = 3, print = TRUE,
  ...)

Arguments

x

A data frame from get_variable_importance

title

Either "model", "none", or a string to be used as the plot caption. "model" puts the name of the best-performing model, on which variable importances are generated, in the title.

max_char

Maximum length of variable names to leave untruncated. Default = 40; use Inf to prevent truncation. Variable names longer than this will be truncated to leave the beginning and end of each variable name, bridged by " ... ".

caption

Plot title

font_size

Relative size for all fonts, default = 11

point_size

Size of dots, default = 3

print

Print the plot?

...

Unused

Value

A ggplot object, invisibly.

Examples

# NOT RUN {
machine_learn(pima_diabetes[1:50, ], patient_id, outcome = diabetes, tune = FALSE) %>%
  get_variable_importance() %>%
  plot()
# }