rbartpackages.BART3

Wrapper for the R package BART3 (rsparapa/bnptools, BART3 subdirectory).

Model fitting

gbart(x_train, y_train[, x_test, type, ...])

Fit BART to continuous or binary outcomes with a single MCMC chain.

mc_gbart(x_train, y_train[, x_test, type, ...])

Fit BART to continuous or binary outcomes with multiple MCMC chains.

Data preprocessing

bartModelMatrix(X[, numcut, usequants, ...])

Convert covariates to a matrix and compute the BART cutpoints.

Supporting types

TreeDraws

Type of the mc_gbart.treedraws attribute.

PredictBinary

Type of mc_gbart.predict's return value for binary ('pbart'/'lbart') fits.

ProcTime(user_self, sys_self, elapsed, ...)

Python representation of the output of R's proc.time.