[英]Grouped linear regression prediction on different grouped by group in R
我正在尝试基于数据集中的特定组构建模型,并使用生成的模型通过遵循组限制来预测不同数据集的拟合。 换句话说,使用下面的示例,使用原始数据的子集:cyl==4 构建的模型应仅用于预测新数据集(data1)的子集:cyl==4。 任何人都可以帮助解决这个有趣的问题吗?
我尝试使用data1%>% group_by(cyl)
来指定新数据,但这没有帮助
谢谢
library(broom)
library(dplyr)
library(purrr)
data1 <- head(mtcars,20)
x<-mtcars %>%
group_by(cyl) %>%
summarise(fit = list(lm(wt ~ mpg)),
data = list(cur_data())) %>%
mutate(col = map(fit, augment, newdata = data1%>% group_by(cyl)))```
这是一个快速的方法来做到这一点
library(dplyr)
models = mtcars %>% group_by(cyl) %>% do(model = lm(wt ~ mpg, data = .))
然后访问各个模型
library(broom)
tidy(models$model[[1]])
做同样的另一种方法 -
models <- mtcars %>%
nest_by(cyl) %>%
mutate(mod = list(lm(mpg ~ disp, data = data)))
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.