abart.fix: fixes bug in abart program from the BART package

In ericdunipace/CoarsePosteriorSummary: Interpretable Model Summaries Using the Wasserstein Distance

Description

fixes bug in abart program from the BART package

Usage

abart.fix(
  x.train,
  times,
  delta,
  x.test = matrix(0, 0, 0),
  K = 100,
  type = "abart",
  ntype = 1,
  sparse = FALSE,
  theta = 0,
  omega = 1,
  a = 0.5,
  b = 1,
  augment = FALSE,
  rho = NULL,
  xinfo = matrix(0, 0, 0),
  usequants = FALSE,
  rm.const = TRUE,
  sigest = NA,
  sigdf = 3,
  sigquant = 0.9,
  k = 2,
  power = 2,
  base = 0.95,
  lambda = NA,
  tau.num = c(NA, 3, 6)[ntype],
  offset = NULL,
  w = rep(1, length(times)),
  ntree = c(200L, 50L, 50L)[ntype],
  numcut = 100L,
  ndpost = 1000L,
  nskip = 100L,
  keepevery = c(1L, 10L, 10L)[ntype],
  printevery = 100L,
  transposed = FALSE,
  mc.cores = 1L,
  nice = 19L,
  seed = 99L
)

Details

For arguments and such see abart.

ericdunipace/CoarsePosteriorSummary documentation built on April 3, 2021, 8:49 p.m.