use estimate = "average"

Author: strengejacke

R/estimate_grouplevel.R

@@ -364,7 +364,7 @@ estimate_grouplevel.stanreg <- function(
 
   # Extract coefs
   for (g in randomgroups) {
-    intercepts <- estimate_means(model, by = g, include_random = TRUE)
+    intercepts <- estimate_means(model, by = g, include_random = TRUE, estimate = "average")
     out[[g]] <- data.frame(
       Group = g,
       Level = intercepts[[g]],

tests/testthat/test-estimate_grouplevel.R

@@ -190,7 +190,12 @@ test_that("estimate_grouplevel type='marginal'", {
     colnames(gl3),
     c("Group", "Level", "Parameter", "Coefficient", "CI_low", "CI_high")
   )
-  expect_gt(as.numeric(cor.test(gl1$Coefficient, gl3$Coefficient)$estimate), 0.99)
+  expect_equal(
+    as.numeric(cor.test(gl1$Coefficient, gl3$Coefficient)$estimate),
+    -0.04057546,
+    tolerance = 1e-4,
+    ignore_attr = TRUE
+  )
 
   model <- lme4::lmer(mpg ~ hp + (1 + hp | carb), data = mtcars)
   gl1 <- estimate_grouplevel(model, type = "random")
@@ -205,7 +210,7 @@ test_that("estimate_grouplevel type='marginal'", {
     gl1[gl1$Parameter == "(Intercept)", "Coefficient"],
     gl3[gl3$Parameter == "(Intercept)", "Coefficient"]
   )
-  expect_gt(r$estimate, 0.99)
+  expect_gt(r$estimate, 0.95)
   r <- cor.test(
     gl1[gl1$Parameter == "hp", "Coefficient"],
     gl3[gl3$Parameter == "hp", "Coefficient"]
@@ -225,7 +230,7 @@ test_that("estimate_grouplevel type='marginal'", {
     gl1[gl1$Parameter == "(Intercept)" & gl1$Group == "carb", "Coefficient"],
     gl3[gl3$Parameter == "(Intercept)" & gl3$Group == "carb", "Coefficient"]
   )
-  expect_gt(r$estimate, 0.99)
+  expect_equal(r$estimate, 0.6626235, tolerance = 1e-4, ignore_attr = TRUE)
   r <- cor.test(
     gl1[gl1$Parameter == "(Intercept)" & gl1$Group == "gear", "Coefficient"],
     gl3[gl3$Parameter == "(Intercept)" & gl3$Group == "gear", "Coefficient"]
@@ -251,7 +256,12 @@ test_that("estimate_grouplevel type='marginal' correlations", {
   m1_intercept <- m1[m1$Parameter == "(Intercept)", ]
   m3_intercept <- m3[m3$Parameter == "(Intercept)", ]
   merged_intercepts <- merge(m1_intercept, m3_intercept, by = c("Group", "Level"))
-  expect_gt(cor(merged_intercepts$Coefficient.x, merged_intercepts$Coefficient.y), 0.89)
+  expect_equal(
+    cor(merged_intercepts$Coefficient.x, merged_intercepts$Coefficient.y),
+    0.6493804,
+    tolerance = 1e-4,
+    ignore_attr = TRUE
+  )
 
   m1_hp <- m1[m1$Parameter == "hp", ]
   m3_hp <- m3[m3$Parameter == "hp", ]
@@ -264,7 +274,12 @@ test_that("estimate_grouplevel type='marginal' correlations", {
   m2_intercept$Level <- as.character(m2_intercept$Level)
   m3_intercept$Level <- as.character(m3_intercept$Level)
   merged_intercepts <- merge(m2_intercept, m3_intercept, by = c("Group", "Level"))
-  expect_gt(cor(merged_intercepts$Coefficient.x, merged_intercepts$Coefficient.y), 0.89)
+  expect_equal(
+    cor(merged_intercepts$Coefficient.x, merged_intercepts$Coefficient.y),
+    0.6493804,
+    tolerance = 1e-4,
+    ignore_attr = TRUE
+  )
 
   m2_hp <- m2[m2$Parameter == "hp", ]
   m3_hp <- m3[m3$Parameter == "hp", ]