Skip to content

Commit 834bddc

Browse files
authored
Merge pull request #97 from StochasticTree/sample_tau_hotfix
Remove erroneous reference to sample_tau
2 parents 9d1f7b6 + d66eeb5 commit 834bddc

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

R/bart.R

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,7 @@ bart <- function(X_train, y_train, W_train = NULL, group_ids_train = NULL,
351351
output_dimension = ncol(W_train)
352352
is_leaf_constant = F
353353
leaf_regression = T
354-
if (sample_tau) {
354+
if (sample_sigma_leaf) {
355355
stop("Sampling leaf scale not yet supported for multivariate leaf models")
356356
}
357357
}
@@ -875,7 +875,7 @@ predict.bartmodel <- function(bart, X_test, W_test = NULL, group_ids_test = NULL
875875
#' bart_model <- bart(X_train = X_train, y_train = y_train,
876876
#' group_ids_train = group_ids_train, rfx_basis_train = rfx_basis_train,
877877
#' X_test = X_test, group_ids_test = group_ids_test, rfx_basis_test = rfx_basis_test,
878-
#' num_gfr = 100, num_burnin = 0, num_mcmc = 100, sample_tau = TRUE)
878+
#' num_gfr = 100, num_burnin = 0, num_mcmc = 100)
879879
#' rfx_samples <- getRandomEffectSamples(bart_model)
880880
getRandomEffectSamples.bartmodel <- function(object, ...){
881881
result = list()

man/getRandomEffectSamples.bartmodel.Rd

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)