Использование dplyr внутри функции карты на вложенном фрейме данных - PullRequest
Новая
       15

Использование dplyr внутри функции карты на вложенном фрейме данных

1 голос
/ 16 мая 2019

сопоставление пользовательской функции с вложенным рамкой

Я пытаюсь сопоставить пользовательскую функцию с вложенным рамкой.Эта функция использовала оператор dplyr и pipe "%>%".

Я пытался включить переменные, используемые внутри функции, но она продолжает не работать.Не уверен, как это сделать. 


table <- expand.grid(x = c("bird", "dogs"), year = c(2018,2019), week= c(1:52)) %>% 
  mutate(christmas = case_when(week == 52 ~1, TRUE ~ 0),
         ev = case_when(week == 30 ~ 1, TRUE ~ 0),
         alpha = rnorm(n = 208, mean = 10))

 myfun_model_alpha <- function(time_s, param, yr, wk){

        event <- time_s %>% select(christmas,ev )
        time_s <- ts(time_s$alpha, start = c(min(time_s$year),min(time_s$week[time_s$year == min(time_s$year)])),end =  c(max(time_s$year),max(time_s$week[time_s$year == max(time_s$year)])), frequency = 52)
        #time_s <- ts(time_s$alpha, start = c(2017,01),end =  c(2019,20), frequency = 52)
        ts_vec <- window(time_s, end = c(yr, wk))
        leng <- length(ts_vec)
        lambda <- BoxCox.lambda(ts_vec)
        model <-  auto.arima(ts_vec, lambda = lambda, 
                             biasadj = TRUE,
                             xreg =  matrix(c(fourier(ts_vec, K=param), 
                                              event$christmas[1:leng],
                                              event$ev[1:leng]), 
                                            nrow = leng, 
                                            ncol = param*2+2), 
                              seasonal = FALSE)

         forecast <-  forecast(model,
                               16,
                               xreg=matrix(c(fourier(ts_vec, K=param, h=16),
                                          event$christmas[(leng+1):(leng+16)], 
                                          event$ev[(leng+1):(leng+16)]),
                                          nrow = 16, 
                                          ncol = param*2+2))$mean

          pred = tibble(forecast = forecast, 
                        time_index = 1:16,
                        actual = window(time_s, start= c(yr,wk))[2:17])


    return(pred)
  }


# Applying the function t the nested df
table %>% 
  group_by(x) %>% 
  nest() %>% 
  mutate(data = map(data, ~arrange(.x, year, week)),
         model = map(data, ~myfun_model_alpha(.,2,2019, 12))) %>% 
  unnest(model)

Возвращает ошибки типа: «Ошибка в (функция (x): объект« рождество »не найден». Я пытаюсь получить прогноз для горизонта 16 недель с помощьюфактические значения (если есть)

...