The R package PSPI (Propensity Score Predictive Inference) provides a suite of Propensity Score Predictive Inference (PSPI) methods to generalize treatment effects in trials to target populations. The package includes an existing model Bayesian Causal Forest (BCF) and four PSPI models (BCF-PS, FullBART, SplineBART, DSplineBART). These methods leverage Bayesian Additive Regression Trees (BART) to adjust for high-dimensional covariates and nonlinear associations, while SplineBART and DSplineBART further use propensity score based splines to address covariate shift between trial data and target population.
This package is based on Rcpp, RcppArmadillo, RcppDist, and pg, please make sure these three packages can be installed.
This package can be installed from R CRAN:
install.packages("PSPI")
library(PSPI)
or Github:
require("devtools")
install_github("https://github.com/zjg540066169/PSPI")
library(PSPI)
This package includes code derived from the BART3 package, originally developed by Rodney Sparapani.
The original source code, licensed under the GNU General Public License version 2 (GPL-2), has been modified as follows:
- We include part of the C++ code in BART3, primarily about functions about
wbartandcpwart. We also modify some files to make sure our package can be successfully compiled. - Modifications were made by Jungang Zou, 2024.
- The original BART3 package is licensed under the GNU General Public License version 2 (GPL-2).
- This package, as a derived work, is also licensed under the GNU General Public License version 2 (GPL-2) to comply with the licensing terms.
- Zou: Author` last name.
- PSPI: Propensity Score Predictive Inference
- BART: Bayesian Additive Regression Trees.
- BCF: Bayesian Causal Forest.
- BCF-PS: Name of a PSPI model.
- FullBART: Name of a PSPI model.
- SplineBART: Name of a PSPI model.
- DSplineBART: Name of a PSPI model.
- MCMC: Monte Carlo Markov chain.
- Cloglog: complementary log–log transform.
- InvGumbel: Inverse Gumbel function or Gumbel quantile function.