plot_varimp_xgboost {autostats}R Documentation

Plot varimp xgboost

Description

recommended parameters to control;

Usage

plot_varimp_xgboost(
  xgb,
  top_n = 10L,
  aggregate = NULL,
  as_table = FALSE,
  formula = NULL,
  measure = c("Gain", "Cover", "Frequency"),
  ...
)

Arguments

xgb

xgb.Booster model

top_n

top n important variables

aggregate

a character vector. Predictors containing the string will be aggregated, and renamed to that string.

as_table

logical, default FALSE. If TRUE returns importances in a data frame

formula

formula for the model. Use to provide original names if xgboost is scrambling the names internally

measure

choose between Gain, Cover, or Frequency for xgboost importance measure

...

additional arguments for xgb.ggplot.importance

Details

top_n

number of features to include in the graph

Value

ggplot


[Package autostats version 0.4.1 Index]