r - ggplot 和 holt winters 预测

Question

使用数据 UKDriverDeaths

尝试使用 Holt-Winters 预测函数和 ggplot()。

基本上重现了ggplot中的数据。

data('UKDriverDeaths')    
past <- window(UKDriverDeaths, end = c(1982, 12))
hw <- HoltWinters(past)
pred <- predict(hw, n.ahead = 10)
plot(hw, pred, ylim = range(UKDriverDeaths))
lines(UKDriverDeaths)

我想展示 holt winters 对 1983 年开始的实际数据的预测。这两个问题是:

1) ggplot 不理解 ts 数据。

2) 使用 HoltWinters() 使用 ts 数据，而不是 zoo(日期或 xts)。我需要显示切点处的预测和实际数据(通常 + geom_line(aes()) 这样做)

如果置信区间是可能的，那就太好了。

非常感谢，绝对卡住了

Answer 1

我正在使用 xts 自动合并数据。

library(xts)
ts_pred <- ts(c(hw$fitted[, 1], pred), start = 1970, frequency = 12)
df <- merge(as.xts(ts_pred), as.xts(UKDriverDeaths))
names(df) <- c("predicted", "actual")
ggplot(df, aes(x=as.POSIXct(index(df)))) + 
  geom_line(aes(y=predicted), col='red') + 
  geom_line(aes(y=actual), col='black') + 
  theme_bw() +
  geom_vline(xintercept=as.numeric(as.POSIXct("1982-12-01")), linetype="dashed") + 
  labs(title="Holt-Winters filtering\n", x="Time", y="Observed / Fitted") + 
  theme(plot.title = element_text(size=18, face="bold"))