【问题标题】:Current and future prediction in a single data frame in rr中单个数据帧中的当前和未来预测
【发布时间】:2021-06-09 19:18:37
【问题描述】:

我有一个下面的数据集,我想对其进行线性回归,然后在单个数据框中获取它们的拟合值和上限和下限。

该数据集包含每个国家和地区组合的五周数据。即对于 US 和 G 组合线性回归模型将建立,其中金额将在 5 周内按周回归,然后该模型将用于预测 6 周。 我的最终数据框看起来像这样

我所做的是一种组合

data <- data.frame(country = c("US","US","US","US","US","US","US","US","US","US","UK","UK","UK","UK","UK"),
               Area = c("G","G","G","G","G","I","I","I","I","I","A","A","A","A","A"),
               week = c(1,2,3,4,5,1,2,3,4,5,1,2,3,4,5),amount = c(12,23,34,32,12,12,34,45,65,45,45,34,23,43,43))

data_1 <- data[(data$country=="US" & data$Area=="G"),]
# model building on initial five weeks
model <- lm(amount ~ week, data = data_1)


# creating one future week: week no 6
future <- data.frame(country="US",Area="G",week=6,amount = NA)
data2 <- rbind(data_1,future)

# prediction on all 6 weeks
pre <- predict(model,newdata = data2,interval = "prediction",level = 0.95)
pre

我从这里得到了提示Looping linear regression output in a data frame in r

但无法复制此结果以包括所有其他组合的第 6 周预测。 请帮忙!

【问题讨论】:

    标签: r


    【解决方案1】:

    我主要是抄你的应对,最后只是结合结果的问题

    data_1 <- data[(data$country=="US" & data$Area=="G"),]
    # model building on initial five weeks
    model <- lm(amount ~ week, data = data_1)
    pre1 <- predict(model,newdata = data_1, interval = "prediction",level = 0.95)
    
    # creating one future week: week no 6
    future <- data.frame(country="US",Area="G",week=6,amount = NA)
    data_2 <- rbind(data_1,future)
    
    # prediction on all 6 weeks
    pre2 <- predict(model,newdata = data_2, interval = "prediction",level = 0.95)
    
    rbind(
      cbind(data_1,pre1),
      cbind(data_2,pre2)
    )
    
       country Area week amount  fit       lwr      upr
    1       US    G    1     12 20.8 -27.67915 69.27915
    2       US    G    2     23 21.7 -21.99851 65.39851
    3       US    G    3     34 22.6 -19.38417 64.58417
    4       US    G    4     32 23.5 -20.19851 67.19851
    5       US    G    5     12 24.4 -24.07915 72.87915
    11      US    G    1     12 20.8 -27.67915 69.27915
    21      US    G    2     23 21.7 -21.99851 65.39851
    31      US    G    3     34 22.6 -19.38417 64.58417
    41      US    G    4     32 23.5 -20.19851 67.19851
    51      US    G    5     12 24.4 -24.07915 72.87915
    6       US    G    6     NA 25.3 -30.23984 80.83984
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2023-03-21
      • 2020-02-20
      • 2018-07-30
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2019-04-02
      • 2023-02-21
      相关资源
      最近更新 更多