По-перше, майте на увазі, що forecast
обчислюєте вибіркові прогнози, але вас цікавлять спостереження у вибірці.
Фільтр Kalman обробляє відсутні значення. Таким чином, ви можете взяти форму простору стану моделі ARIMA з результатів, повернутих forecast::auto.arima
або stats::arima
передати їх KalmanRun
.
Редагувати (виправити в коді на основі відповіді stats0007)
У попередній версії я взяв стовпчик відфільтрованих станів, що стосується спостережуваного ряду, однак я повинен використовувати всю матрицю і виконувати відповідну матричну операцію рівняння спостереження, . (Спасибі @ stats0007 за коментарі.) Нижче я відповідно оновлюю код та малюнок.ут= Zαт
Я використовую ts
об'єкт як зразок серії замість zoo
, але він повинен бути таким же:
require(forecast)
# sample series
x0 <- x <- log(AirPassengers)
y <- x
# set some missing values
x[c(10,60:71,100,130)] <- NA
# fit model
fit <- auto.arima(x)
# Kalman filter
kr <- KalmanRun(x, fit$model)
# impute missing values Z %*% alpha at each missing observation
id.na <- which(is.na(x))
for (i in id.na)
y[i] <- fit$model$Z %*% kr$states[i,]
# alternative to the explicit loop above
sapply(id.na, FUN = function(x, Z, alpha) Z %*% alpha[x,],
Z = fit$model$Z, alpha = kr$states)
y[id.na]
# [1] 4.767653 5.348100 5.364654 5.397167 5.523751 5.478211 5.482107 5.593442
# [9] 5.666549 5.701984 5.569021 5.463723 5.339286 5.855145 6.005067
Ви можете побудувати результат (для всієї серії та за весь рік із відсутніми спостереженнями в середині вибірки):
par(mfrow = c(2, 1), mar = c(2.2,2.2,2,2))
plot(x0, col = "gray")
lines(x)
points(time(x0)[id.na], x0[id.na], col = "blue", pch = 19)
points(time(y)[id.na], y[id.na], col = "red", pch = 17)
legend("topleft", legend = c("true values", "imputed values"),
col = c("blue", "red"), pch = c(19, 17))
plot(time(x0)[60:71], x0[60:71], type = "b", col = "blue",
pch = 19, ylim = range(x0[60:71]))
points(time(y)[60:71], y[60:71], col = "red", pch = 17)
lines(time(y)[60:71], y[60:71], col = "red")
legend("topleft", legend = c("true values", "imputed values"),
col = c("blue", "red"), pch = c(19, 17), lty = c(1, 1))
Ви можете повторити той же приклад, використовуючи плавніше Кальмана замість фільтра Калмана. Все, що вам потрібно змінити, це ці рядки:
kr <- KalmanSmooth(x, fit$model)
y[i] <- kr$smooth[i,]
Справа з пропущеними спостереженнями за допомогою фільтра Калмана іноді трактується як екстраполяція серії; коли використовується більш гладкий калман, пропущені спостереження, як кажуть, заповнюються інтерполяцією у спостережуваній серії.