add_imp_plot | R Documentation |
adds bar plot of important features to model response alluvial plot
add_imp_plot(grid, p = NULL, data_input, plot = T, ...)
grid |
gtable or ggplot |
p |
alluvial plot, optional if alluvial plot has already been passed as grid. Default: NULL |
data_input |
dataframe used to generate alluvial plot |
plot |
logical if plot should be drawn or not |
... |
additional parameters passed to |
gtable
arrangeGrob
plot_imp
## 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 = 4, pred_train = pred_train)
p_grid = add_marginal_histograms(p, data_input = df)
p_grid = add_imp_plot(p_grid, p, data_input = df)
## End(Not run)
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.