purrr使用模型来数据

时间:2018-03-15 13:39:35

标签: r purrr

我没有看到任何问题。我想以下,当我看到purrr很多模型的例子时,如何再次使用在数据上创建的模型?一点点代码会告诉你我在追求什么:

这是基本的gapminder很多模型的例子。

{{1}}

我认为不是仅仅使用看到的数据拟合值,我可以将模型滞后一个,而我想要做的是用这个滞后模型预测数据。我知道这是一个很糟糕的例子(为什么我会将阿富汗模型用于阿尔巴尼亚),但我有数据与日期嵌套,并且它会有意义。这仍然是可重复的例子。那么在这种格式中使用预测呢?

结果将是新列“pred”[12x1],对新数据进行预测,然后我可以用数据取消它并在那里进行预测。

1 个答案:

答案 0 :(得分:1)

您应该可以使用purrr::map2执行此操作。我不太确定如何评估NA值,所以我只是过滤掉它。

library(gapminder)

gapminder %>% 
  group_by(country, continent) %>% 
  nest() %>%
  mutate(model = map(data, partial(lm, lifeExp ~ year))) %>% 
  mutate(model_lag = lag(model)) %>%
  filter(!is.na(model_lag)) %>%
  mutate(pred = map2(model_lag, data, predict))