簡體   English   中英

R中lm對象的預測函數

[英]Predict function for lm object in R

為什么prediction_meprediction_R不相等? 我試圖遵循引理5給出的公式在這里 predict函數是否使用其他公式,是否在某個地方的計算中犯了錯誤,還是僅舍入了誤差? (兩者非常接近)

set.seed(100)
# genrate data
x    <- rnorm(100, 10)
y    <- 3 + x + rnorm(100, 5)
data <- data.frame(x = x, y = y)
# fit model
mod  <- lm(y ~ x, data = data)

# new observation
data2 <- data.frame(x = rnorm(5, 10))

# prediction for new observation
d    <- as.matrix(cbind(1, data[,-2]))
d2   <- as.matrix(cbind(1, data2))
fit  <- d2 %*% mod$coefficients 
t    <- qt(1 - .025, mod$df.residual)
s    <- summary(mod)$sigma
half <- as.vector(t*s*sqrt(1 + d2%*%solve(t(d)%*%d, t(d2))))

prediction_me <- cbind(fit, fit - half, fit + half)

prediction_R <- predict(mod, newdata = data2, interval = 'prediction')


prediction_me
prediction_R

您當前的代碼幾乎可以了。 只需注意引理5中的公式是針對單個新觀察到的x 因此, half不僅包含相關方差,還包含協方差,而您只需要前一個方差。 因此, as.vector替換為diag

half <- diag(t * s * sqrt(1 + d2 %*% solve(t(d) %*%d , t(d2))))
prediction_me <- cbind(fit, fit - half, fit + half)
prediction_R <- predict(mod, newdata = data2, interval = 'prediction')

range(prediction_me - prediction_R)
# [1] 0 0

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM