Пользовательская функция R lm в тидиверсе? - PullRequest
Новая
       9

Пользовательская функция R lm в тидиверсе?

1 голос
/ 26 апреля 2020

У меня простой вопрос о создании пользовательских функций lm в рамках tidyverse.

Мне в основном нужна функция, которая запускает пользовательскую модель со мной с одной свободной переменной. 

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

А потом я хочу использовать это в dplyr do

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

Как мне подойти к этой проблеме?

1 Ответ

1 голос
/ 26 апреля 2020

Вы можете передать данные в функцию: 

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>
...