R/helper_get_priors.R

In RSTr: Gibbs Samplers for Discrete Bayesian Spatiotemporal Models

get_priors <- function(RSTr_obj, priors) {
  UseMethod("get_priors")
}

#' @export
get_priors.car <- function(RSTr_obj, priors) {
  priors$lambda_sd <- priors$lambda_sd %||%
    array(0.025, dim = dim(RSTr_obj$data$Y))
  priors$lambda_acpt <- array(0, dim = dim(priors$lambda_sd))
  priors$tau_a <- priors$tau_a %||% 0.001
  priors$tau_b <- priors$tau_b %||% 0.001
  priors$sig_a <- priors$sig_a %||% 0.001
  priors$sig_b <- priors$sig_b %||% 0.001
  RSTr_obj$priors <- priors
  RSTr_obj
}

#' @export
get_priors.mcar <- function(RSTr_obj, priors) {
  n_group <- dim(RSTr_obj$data$Y)[2]
  priors$lambda_sd <- priors$lambda_sd %||%
    array(0.025, dim = dim(RSTr_obj$data$Y))
  priors$lambda_acpt <- array(0, dim = dim(priors$lambda_sd))
  priors$tau_a <- priors$tau_a %||% 0.001
  priors$tau_b <- priors$tau_b %||% 0.001
  priors$G_df <- priors$G_df %||% (n_group + 2)
  priors$G_scale <- priors$G_scale %||% diag(1 / 7, n_group)
  RSTr_obj$priors <- priors
  RSTr_obj
}

#' @export
get_priors.mstcar <- function(RSTr_obj, priors) {
  n_group <- dim(RSTr_obj$data$Y)[2]
  priors$lambda_sd <- priors$lambda_sd %||%
    array(0.025, dim = dim(RSTr_obj$data$Y))
  priors$lambda_acpt <- array(0, dim = dim(priors$lambda_sd))
  priors$tau_a <- priors$tau_a %||% 0.001
  priors$tau_b <- priors$tau_b %||% 0.001
  priors$G_df <- priors$G_df %||% (n_group + 2)
  priors$G_scale <- priors$G_scale %||% diag(1 / 7, n_group)
  priors$Ag_df <- priors$Ag_df %||% (n_group + 2)
  priors$Ag_scale <- priors$Ag_scale %||% diag(1 / 7, n_group)
  RSTr_obj$priors <- priors
  RSTr_obj
}

#' @export
get_priors.mstcar_update_rho <- function(RSTr_obj, priors) {
  n_group <- dim(RSTr_obj$data$Y)[2]
  priors$lambda_sd <- priors$lambda_sd %||%
    array(0.025, dim = dim(RSTr_obj$data$Y))
  priors$lambda_acpt <- array(0, dim = dim(priors$lambda_sd))
  priors$tau_a <- priors$tau_a %||% 0.001
  priors$tau_b <- priors$tau_b %||% 0.001
  priors$G_df <- priors$G_df %||% (n_group + 2)
  priors$G_scale <- priors$G_scale %||% diag(1 / 7, n_group)
  priors$Ag_scale <- priors$Ag_scale %||% diag(1 / 7, n_group)
  priors$Ag_df <- priors$Ag_df %||% (n_group + 2)
  priors$rho_a <- priors$rho_a %||% 95
  priors$rho_b <- priors$rho_b %||% 5
  priors$rho_sd <- priors$rho_sd %||% matrix(0.05, n_group, 1)
  priors$rho_acpt <- array(0, dim = dim(priors$rho_sd))
  RSTr_obj$priors <- priors
  RSTr_obj
}