R/predictive_plot_simple.R

In igutierrezm/ANOVABNPTestR: A Bayesian Nonparametric Factorial ANOVA

#' #' Plot the posterior predictive pdf (simple effects)
#' #'
#' #' Plot the posterior predictive pdf for each value of 1 factor,
#' #' fixing the others at the reference level.
#' #' @param fit An object of class anova_bnp_model.
#' #' @param d1 The factor id.
#' #' @return A ggplot2 plot.
#' #' @importFrom dplyr mutate filter left_join
#' #' @importFrom dplyr c_across across pull rowwise dense_rank
#' #' @importFrom ggplot2 ggplot aes_string geom_line
#' #' @importFrom rlang := .data
#' #' @export
#' predictive_plot_simple <- function(fit, d1) {
#'   # Get the relevant groups
#'   var1 <- paste0("x", d1)
#'   target_groups <-
#'     group_codes(fit) |>
#'     mutate(across(-c("group"), ~ dense_rank(.x))) |>
#'     rowwise() |>
#'     mutate(touse = max(c_across(-c("group", var1)))) |>
#'     filter(.data$touse == 1) |>
#'     pull("group")
#'
#'   # Plot the posterior predictive pdf
#'   group_codes(fit) |>
#'     filter(.data$group %in% target_groups) |>
#'     left_join(f_post(fit)) |>
#'     mutate({{ var1 }} := factor(.data[[var1]])) |>
#'     ggplot(aes_string(x = "y", y = "f", color = var1)) +
#'     geom_line()
#' }