easyalluvial allows you to build exploratory alluvial plots (sankey diagrams) with a single line of code while automatically binning numerical variables. This releas
tidyr 1.0.0 compatibility and fixes a bug around categorical variables for model response plots
Model Response Plots with Categorical Variables
This feature had som glitches before as edvardoss reported in this issue. If anybody else encounters some glitches or inconcistencies please report them as well.
We create a grid of all possible feature combinations and use an alluvial plot to visualise the model response.
Learn more about this feature in this previous blog post
suppressPackageStartupMessages( require(tidyverse) ) suppressPackageStartupMessages( require(easyalluvial) ) df = titanic %>% select_if(is.factor) set.seed(0) m = randomForest::randomForest( Survived ~ ., df) imp = m$importance dspace = get_data_space(df, imp, degree = 3) pred = predict(m, newdata = dspace,type = 'response') p = alluvial_model_response(pred, dspace, imp, degree = 3) grid = add_marginal_histograms(p, plot = F, data_input = df) grid = add_imp_plot(grid = grid, p = p, data_input = df, plot = T)