From d66eeb53af45c66312cd69941ec2f47208b1e43b Mon Sep 17 00:00:00 2001 From: Drew Herren Date: Sun, 13 Oct 2024 21:59:42 -0500 Subject: [PATCH] Remove dependence on sample_tau --- R/bart.R | 4 ++-- man/getRandomEffectSamples.bartmodel.Rd | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/R/bart.R b/R/bart.R index 47afeede..a3e70bcd 100644 --- a/R/bart.R +++ b/R/bart.R @@ -351,7 +351,7 @@ bart <- function(X_train, y_train, W_train = NULL, group_ids_train = NULL, output_dimension = ncol(W_train) is_leaf_constant = F leaf_regression = T - if (sample_tau) { + if (sample_sigma_leaf) { stop("Sampling leaf scale not yet supported for multivariate leaf models") } } @@ -875,7 +875,7 @@ predict.bartmodel <- function(bart, X_test, W_test = NULL, group_ids_test = NULL #' bart_model <- bart(X_train = X_train, y_train = y_train, #' group_ids_train = group_ids_train, rfx_basis_train = rfx_basis_train, #' X_test = X_test, group_ids_test = group_ids_test, rfx_basis_test = rfx_basis_test, -#' num_gfr = 100, num_burnin = 0, num_mcmc = 100, sample_tau = TRUE) +#' num_gfr = 100, num_burnin = 0, num_mcmc = 100) #' rfx_samples <- getRandomEffectSamples(bart_model) getRandomEffectSamples.bartmodel <- function(object, ...){ result = list() diff --git a/man/getRandomEffectSamples.bartmodel.Rd b/man/getRandomEffectSamples.bartmodel.Rd index 3969ae49..72419b54 100644 --- a/man/getRandomEffectSamples.bartmodel.Rd +++ b/man/getRandomEffectSamples.bartmodel.Rd @@ -54,6 +54,6 @@ rfx_term_train <- rfx_term[train_inds] bart_model <- bart(X_train = X_train, y_train = y_train, group_ids_train = group_ids_train, rfx_basis_train = rfx_basis_train, X_test = X_test, group_ids_test = group_ids_test, rfx_basis_test = rfx_basis_test, - num_gfr = 100, num_burnin = 0, num_mcmc = 100, sample_tau = TRUE) + num_gfr = 100, num_burnin = 0, num_mcmc = 100) rfx_samples <- getRandomEffectSamples(bart_model) }