Exemplo de agrupar forecasts

exemplo_curso_st.R

library(tidyverse)
library(purrr)
library(forecast)
library(lubridate)

vetor_uf <- c("RJ", "SP", "MG", "ES")

vetor_data <- seq.Date(as.Date("2010-01-01"), by = "month",
                       length.out = 60)

df <- data.frame(
  uf = rep(vetor_uf, each = 12 * 5),
  valor = rnorm(12 * 5 * 4, mean = 100, sd = 5),
  data = rep(vetor_data, 4)
)

ggplot(df, aes(x = data, y = valor)) +
  geom_line() +
  facet_wrap(~ uf)

extrair_previsao <- function(dframe){
  # data: dataframe separado por UF
  # construir objeto ts
  st_ts_uf <- decimal_date(min(dframe[["data"]]))
  ts_uf <- ts(dframe$valor, start = st_ts_uf, frequency = 12)
  # ajustar um modelo sazonal ingenuo
  mod_saz <- snaive(ts_uf)
  # retornar previsao obtida pelo modelo
  mod_saz$mean
}

lista_uf <- df %>% 
  split(.$uf)

lista_uf %>% 
  map(extrair_previsao)