tidyverse 中的自定义 R lm 函数？

Question

我有一个关于在 tidyverse 框架内创建自定义 lm 函数的简单问题。

我基本上想要一个函数，它可以使用一个自由变量与我一起运行自定义模型。

model <- function(x){
  lmer(paste("cyl ~", x, "+ (1|disp)"), data = .)
}

然后我想在 dplyr 的 do

中使用它

mtcars %>% 
  do(x = model("hp"))

我该如何解决这个问题？

Answer 1

您可以将数据传递给函数：

library(dplyr)
library(lme4)

model <- function(data, x){
  lmer(paste("cyl ~", x, "+", "(1|disp)"), data = data)
}

然后像这样称呼它：

mtcars %>% model('hp')

#Linear mixed model fit by REML ['lmerMod']
#Formula: cyl ~ hp + (1 | disp)
#   Data: data
#REML criterion at convergence: 96.2
#Random effects:
# Groups   Name        Std.Dev.
# disp     (Intercept) 0.927   
# Residual             0.441   
#Number of obs: 32, groups:  disp, 27
#Fixed Effects:
#(Intercept)           hp  
#     3.1866       0.0196

或者

mtcars %>% summarise(mod = list(model(., 'hp')))

#                                                  mod
#1 <S4 class ‘lmerMod’ [package “lme4”] with 13 slots>

tidyverse 中的自定义 R lm 函数？

Custom R lm function within the tidyverse?

r

lme4

dplyr