R/extractPred.R

In abnormally-distributed/Bayezilla: Bayesian Models With JAGS and Several Utilities for Data Exploration and Model Evaluation.

#' Extract predicted y values (posterior predictions)
#'
#' @param x the stanfit or runjags object containing posterior predictions
#' @param predname default name is "ySim".
#' @export
#' @examples
#' extractPred()
#'
extractPred =function (x, predname = "ySim")
{
  stan <- inherits(x, "stanfit")
  if (stan == TRUE) {
    ss <- as.matrix(x, pars = predname)
  }
  else if (class(x) == "runjags"){
    ss <- runjags::combine.mcmc(x, collapse.chains = TRUE, vars = predname)
    ss <- as.matrix(ss)
  }

  df = as.matrix(ss)
  rownames(df) = c(paste0("yIter[", 1:nrow(df), "]"))
  return(df)
}