R/get_draws.R

In n-kall/priorsense: Prior Diagnostics and Sensitivity Analysis

get_draws_stanfit <- function(x, variable = NULL,
                              excluded_variables = c(
                                "lprior", "lp__", "log_lik"
                              ),
                              ...) {
  if (is.null(variable)) {
    draws <- remove_unwanted_vars(x)
  } else {
    draws <- posterior::as_draws_df(as.array(x))
  }

  return(draws)
}

get_draws_CmdStanFit <- function(x, variable = NULL, regex,
                                 excluded_variables = c(
                                   "lprior", "lp__", "log_lik"
                                 ),
                                 ...) {

  if (is.null(variable)) {
    draws <- remove_unwanted_vars(x)
  } else {
    draws <- posterior::as_draws_df(x$draws())
  }
  return(draws)
}