plot_imp: plot feature importance

Description Usage Arguments Value Examples

View source: R/plot_imp.R

Description

plot important features of model response alluvial as bars

Usage

1
plot_imp(p, data_input, truncate_at = 50, color = "darkgrey")

Arguments

p

alluvial plot

data_input

dataframe used to generate alluvial plot

truncate_at

integer, limit number of features to that value, Default: 50

color

character vector, Default: 'darkgrey'

Value

ggplot object

Examples

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
## Not run: 
df = mtcars2[, ! names(mtcars2) %in% 'ids' ]

train = caret::train( disp ~ .
                     , df
                     , method = 'rf'
                     , trControl = caret::trainControl( method = 'none' )
                     , importance = TRUE )

pred_train = caret::predict.train(train, df)

p = alluvial_model_response_caret(train, degree = 3, pred_train = pred_train)

plot_imp(p, mtcars2)


## End(Not run)

erblast/easyalluvial documentation built on Aug. 8, 2020, 1:11 a.m.