|
98 | 98 | #' ) |
99 | 99 | #' |
100 | 100 | #' # add ellipse |
101 | | -#' p + stat_ellipse(level = 0.9, color = "gray20", size = 1) |
| 101 | +#' p + stat_ellipse(level = 0.9, color = "gray20", linewidth = 1) |
102 | 102 | #' |
103 | 103 | #' # add contour |
104 | 104 | #' color_scheme_set("red") |
105 | 105 | #' p2 <- mcmc_scatter(x, pars = c("alpha", "sigma"), size = 3.5, alpha = 0.25) |
106 | | -#' p2 + stat_density_2d(color = "black", size = .5) |
| 106 | +#' p2 + stat_density_2d(color = "black", linewidth = .5) |
107 | 107 | #' |
108 | 108 | #' # can also add lines/smooths |
109 | 109 | #' color_scheme_set("pink") |
@@ -352,10 +352,10 @@ mcmc_pairs <- function(x, |
352 | 352 | param <- sym("Parameter") |
353 | 353 | val <- sym("Value") |
354 | 354 | np <- validate_nuts_data_frame(np, lp) |
355 | | - divs <- dplyr::filter(np, UQ(param) == "divergent__") %>% pull(UQ(val)) |
| 355 | + divs <- dplyr::filter(np, !!param == "divergent__") %>% pull(!!val) |
356 | 356 | divergent__ <- matrix(divs, nrow = n_iter * n_chain, ncol = n_param)[, 1] |
357 | 357 | if (!no_max_td) { |
358 | | - gt_max_td <- (dplyr::filter(np, UQ(param) == "treedepth__") %>% pull(UQ(val))) >= max_treedepth |
| 358 | + gt_max_td <- (dplyr::filter(np, !!param == "treedepth__") %>% pull(!!val)) >= max_treedepth |
359 | 359 | max_td_hit__ <- matrix(gt_max_td, nrow = n_iter * n_chain, ncol = n_param)[, 1] |
360 | 360 | } |
361 | 361 | } |
@@ -674,11 +674,11 @@ pairs_condition <- function(chains = NULL, draws = NULL, nuts = NULL) { |
674 | 674 | divg <- sym("Divergent") |
675 | 675 | xydata$Divergent <- |
676 | 676 | np %>% |
677 | | - dplyr::filter(UQ(param) == "divergent__") %>% |
678 | | - pull(UQ(val)) |
| 677 | + dplyr::filter(!!param == "divergent__") %>% |
| 678 | + pull(!!val) |
679 | 679 |
|
680 | | - divdata <- dplyr::filter(xydata, UQ(divg) == 1) |
681 | | - xydata <- dplyr::filter(xydata, UQ(divg) == 0) |
| 680 | + divdata <- dplyr::filter(xydata, !!divg == 1) |
| 681 | + xydata <- dplyr::filter(xydata, !!divg == 0) |
682 | 682 | } |
683 | 683 |
|
684 | 684 | graph <- ggplot(data = xydata, aes(x = .data$x, y = .data$y)) + |
@@ -880,7 +880,7 @@ handle_condition <- function(x, condition=NULL, np=NULL, lp=NULL) { |
880 | 880 |
|
881 | 881 | } else { |
882 | 882 | param <- sym("Parameter") |
883 | | - mark <- dplyr::filter(np, UQ(param) == condition) |
| 883 | + mark <- dplyr::filter(np, !!param == condition) |
884 | 884 | mark <- unstack_to_matrix(mark, Value ~ Chain) |
885 | 885 | } |
886 | 886 | if (condition == "divergent__") { |
|
0 commit comments