浏览代码

handle repeated days for a holiday in R

Ben Letham 8 年之前
父节点
当前提交
b7a201086d
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      R/R/prophet.R

+ 2 - 1
R/R/prophet.R

@@ -9,7 +9,7 @@
 globalVariables(c(
   "ds", "y", "cap", ".",
   "component", "dow", "doy", "holiday", "holidays", "holidays_lower", "holidays_upper", "ix",
-  "lower", "n", "stat", "trend",
+  "lower", "n", "stat", "trend", "row_number",
   "trend_lower", "trend_upper", "upper", "value", "weekly", "weekly_lower", "weekly_upper",
   "x", "yearly", "yearly_lower", "yearly_upper", "yhat", "yhat_lower", "yhat_upper"))
 
@@ -323,6 +323,7 @@ make_holiday_features <- function(m, dates) {
   wide <- m$holidays %>%
     dplyr::mutate(ds = zoo::as.Date(ds)) %>%
     dplyr::group_by(holiday, ds) %>%
+    dplyr::filter(row_number() == 1) %>%
     dplyr::do({
       if (exists('lower_window', where = .) && !is.na(.$lower_window)
           && !is.na(.$upper_window)) {