R/bestan_plot.R

In mvuorre/bestan: Bayesian estimation for the t-test using Stan

bestan_plot <- function(stanobj, color="purple"){

    library(reshape2)
    library(ggplot2)
    library(dplyr)
    # Parameter posterior distribution
    bestan_df <- as.data.frame(stanobj) %>%
        select(-matches("lp__")) %>%
        mutate(nu_log10 = log10(nu))

    # Plot posterior medians and 2.5 & 97.5 percentiles
    ggplot(melt(bestan_df), aes(x=value)) +
        stat_bin(aes(y=..ndensity..),
                 col="white", fill=color) +
        labs(x="", y="") +
        facet_wrap(~variable,
                   scales = "free",
                   ncol=3) +
        theme_minimal() +
        labs(title="Posterior samples of bestan fit") +
        theme(axis.text.y = element_blank(),
              axis.ticks.y = element_blank(),
              panel.background = element_rect(color="gray50"),
              panel.grid = element_blank())
}