bartXViz: Visualization of BART and BARP using SHAP

Complex machine learning models are often difficult to interpret. Shapley values serve as a powerful tool to understand and explain why a model makes a particular prediction. This package computes variable contributions using permutation-based Shapley values for Bayesian Additive Regression Trees (BART) and its extension with Post-Stratification (BARP). The permutation-based SHAP method proposed by Strumbel and Kononenko (2014) <doi:10.1007/s10115-013-0679-x> is grounded in data obtained via MCMC sampling. Similar to the BART model introduced by Chipman, George, and McCulloch (2010) <doi:10.1214/09-AOAS285>, this package leverages Bayesian posterior samples generated during model estimation, allowing variable contributions to be computed without requiring additional sampling. The BART model is designed to work with the following R packages: 'BART' <doi:10.18637/jss.v097.i01>, 'bartMachine' <doi:10.18637/jss.v070.i04>, and 'dbarts' <https://CRAN.R-project.org/package=dbarts>. For XGBoost and baseline adjustments, the approach by Lundberg et al. (2020) <doi:10.1038/s42256-019-0138-9> is also considered. The BARP model proposed by Bisbee (2019) <doi:10.1017/S0003055419000480> was implemented with reference to <https://github.com/jbisbee1/BARP> and is designed to work with modified functions based on that implementation. BARP extends post-stratification by computing variable contributions within each stratum defined by stratifying variables. The resulting Shapley values are visualized through both global and local explanation methods.

Version: 1.0.8
Depends: R (≥ 3.5.0), SuperLearner
Imports: bartMachine, BART, ggplot2, ggforce, data.table, ggfittext, ggpubr, foreach, gggenes, Rcpp, dplyr, tidyr, stringr, abind, utils, grid, dbarts, forcats, gridExtra, reshape2, missForest
LinkingTo: Rcpp, RcppArmadillo
Published: 2025-07-28
DOI: 10.32614/CRAN.package.bartXViz
Author: Dong-eun Lee [aut, cre], Eun-Kyung Lee [aut]
Maintainer: Dong-eun Lee <ldongeun.leel at gmail.com>
License: GPL-2 | GPL-3 [expanded from: GPL (≥ 2)]
NeedsCompilation: yes
CRAN checks: bartXViz results

Documentation:

Reference manual: bartXViz.html , bartXViz.pdf

Downloads:

Package source: bartXViz_1.0.8.tar.gz
Windows binaries: r-devel: bartXViz_1.0.8.zip, r-release: bartXViz_1.0.6.zip, r-oldrel: bartXViz_1.0.6.zip
macOS binaries: r-release (arm64): bartXViz_1.0.8.tgz, r-oldrel (arm64): bartXViz_1.0.8.tgz, r-release (x86_64): bartXViz_1.0.8.tgz, r-oldrel (x86_64): bartXViz_1.0.8.tgz
Old sources: bartXViz archive

Linking:

Please use the canonical form https://CRAN.R-project.org/package=bartXViz to link to this page.