plot_importance: Plot variable importance

In jiho/joml: Utilities for Machine Learning

Plot variable importance

Description

Plot variable importance

Usage

plot_importance(object)

Arguments

object

an object output by importance(), which contains an importance column.

Value

A ggplot2 object.

See Also

Other variable importance functions: importance(), summarise_importance()

Examples

# fit a model on five bootstraps
m <- resample_boot(mtcars, 5) %>%
  xgb_fit(resp="mpg", expl=c("cyl", "hp", "qsec"),
    eta=0.1, max_depth=4, nrounds=20)
# compute variable importance for each model
m <- importance(m)
# look at importance for the first resample
m$importance[[1]]
# summarise across resamples
summarise_importance(m)
# plot the summarised importance
plot_importance(m)

jiho/joml documentation built on Dec. 6, 2023, 5:50 a.m.