R/plot_trace.R

In dreamer: Dose Response Models for Bayesian Model Averaging

#' @title Traceplots
#' @description Produces traceplots for each parameter for each model.
#' @param x output from a call to `dreamer_mcmc()`.
#' @return No return value, called to create plots.
#' @example man/examples/ex-plot_trace.R
#' @export
plot_trace <- function(x) {
  UseMethod("plot_trace", x)
}

#' @export
plot_trace.dreamer_mcmc <- function(x) { #nolint
  old_par <- graphics::par(no.readonly = TRUE)
  on.exit(graphics::par(old_par))
  graphics::par(mfrow = c(3, 2))
  coda::traceplot(x)
}

#' @export
plot_trace.dreamer_bma <- function(x) { #nolint
  ind <- vapply(
    x,
    function(model) inherits(model, "mcmc.list"),
    logical(1)
  ) %>%
    which()
  old_par <- graphics::par(no.readonly = TRUE)
  on.exit(graphics::par(old_par))
  graphics::par(mfrow = c(3, 2))
  for (i in ind) {
    for (j in seq_len(length(x[[i]]))) {
      colnames(x[[i]][[j]]) <- paste0(names(x)[i], "_", colnames(x[[i]][[j]]))
    }
    coda::traceplot(x[[i]])
  }
}