From 0a87fef33554a7b2bd4eb074f1846566eea7e9ca Mon Sep 17 00:00:00 2001 From: Peter Date: Mon, 3 Jun 2024 16:07:47 -0700 Subject: [PATCH] fixes bug in compareEstimates with median estimates --- R/smoothArea.R | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/R/smoothArea.R b/R/smoothArea.R index b54958c..c44f8cc 100644 --- a/R/smoothArea.R +++ b/R/smoothArea.R @@ -426,6 +426,7 @@ plot.svysae <- function(x, return_list = F, plot.factor = NULL, ...) { #' #' @param x svysae object. Plots are created for all models in this object. #' @param posterior.sample Matrix of posteriors samples of area level quantities with one row for each area and one column for each sample. This argument may be specified to only provide a heatmap for the desired samples. +#' @param title Optional parameter changing the title of the plot #' #' @return ggplot containing heat map of pairwise comparisons #' @@ -452,7 +453,8 @@ plot.svysae <- function(x, return_list = F, plot.factor = NULL, ...) { #' compareEstimates(cts.res) #' } compareEstimates <- function(x, - posterior.sample = NULL) { + posterior.sample = NULL, + title = NULL) { x_att <- attributes(x) @@ -488,7 +490,7 @@ compareEstimates <- function(x, Domain1 = x_att$domain.names[median_order], Domain2 = "Median", Prob = NA, - est = domain_medians + est = domain_medians[median_order] ) extra_cols <- c("", "Median", "Interval") @@ -525,9 +527,13 @@ compareEstimates <- function(x, ggplot2::scale_fill_viridis_c(name = "Probability of Domain 1 > Domain 2", na.value = "white") + ggplot2::scale_color_manual(values = c("white", "black"), guide = "none") + - ggplot2::labs(title = x_att$inla.fitted[i], - x = "Domain 2", + ggplot2::labs(x = "Domain 2", y = "Domain 1") + if (!is.null(title)) { + g_heat <- g_heat + ggplot2::labs(title = title) + } else if (is.null(posterior.sample)) { + g_heat <- g_heat + ggplot2::labs(title = x_att$inla.fitted[i]) + } suppressWarnings(print(g_heat)) } }