使用lm()和predict()按日期滚动回归和预测

问题描述 投票:0回答:1

所以我正在使用lm()和predict()进行滚动回归,基于以下问题+最佳答案,它与我的数据完美配合

https://stackoverflow.com/a/38041406/9932223

我的问题是我不想在每一行之后使用新的lm()。我的数据每个日期都有多行,并且每个日期的行数并不总是相同。如果我想在每个日期的行集结束时使用新的lm()(并继续使用之前所有日期的所有先前数据),我将如何修改此代码?

r loops regression lm predict
1个回答
0
投票

考虑在lapply调用中调用方法来迭代lm调用中用于数据帧过滤的日期的唯一值:

# ORDER BY DATE ASC
dat <- with(dat, dat[order(Date),])

bundle <- function(curr_date) {
  # REPLACING subset WITH FILTER FOR ALL DATES BEFORE CURRENT DATE
  fit <- lm(y ~ x1 + x2 + x3, data = dat[dat$Date < curr_date,], model = FALSE)

  # REPLACE FILTER FOR ALL DATES ON CURRENT DATE
  pred <- predict(fit, newdata = dat[dat$Date == curr_date,], se.fit = TRUE)

  # RETURN DATA FRAME OF RESULTS
  data.frame(date = curr_date,
             adj_r = summary(fit)$adj.r.squared, 
             fit = pred$fit, 
             se_fit = pred$se.fit)
}

# LAPPLY CALL 
rolling_models_df_list <- lapply(unique(dat$Date)[-1], bundle)
© www.soinside.com 2019 - 2024. All rights reserved.