每组建立多个模型

时间:2017-07-06 03:55:34

标签: r dplyr tidyr lm broom

我想对我的数据进行分组,然后为每个组构建两个线性模型,收集结果,并使用扫帚来总结模型参数,但我有一个无法理解的无限递归错误。这是代码:

library(dplyr)
library(tidyr)
library(broom)

mtcars %>% 
group_by(am) %>% 
dplyr::do(simple_fit =  lm(mpg ~ disp, data = .), 
          complex_fit = lm(mpg ~ disp + hp, data = .)) %>% 
ungroup()
gather(model_type, model, -am) %>% 
broom::tidy(model)

会导致此错误:

Error: evaluation nested too deeply: infinite recursion / options(expressions=)?

这个例子中只有4个模型,所以我不明白为什么我要打这么深的嵌套循环?

1 个答案:

答案 0 :(得分:2)

我在github上发现了一条修复我的问题here

的评论

固定版本的代码如下:

mtcars %>% 
group_by(am) %>% 
dplyr::do(simple_fit = lm(mpg~disp, data = .), 
          complex_fit = lm(mpg ~ disp + hp, data = .)) %>% 
ungroup() %>% 
gather(model_type, model, -am) %>% 
rowwise() %>% 
broom::tidy(model)