saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-09-16 22:20:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

issue-31-34 patch holiday forecast comp

Browse source
This commit is contained in:
Rob Schoenbeck 2017-02-27 13:35:29 -08:00
parent 1f46ef73a6
commit 1c01713f8d
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
R/R/prophet.R
View file

@ -951,13 +951,13 @@ prophet_plot_components <- function(m, fcst, uncertainty = TRUE) {
panels <- list(gg.trend) panels <- list(gg.trend)
# Plot holiday components, if present. # Plot holiday components, if present.
if (!is.null(m$holidays)) { if (!is.null(m$holidays)) {
holiday.comps <- unique(m$holidays$holiday) holiday.comps <- unique(m$holidays$holiday) %>% as.character()
df.s <- data.frame(ds = df$ds, df.s <- data.frame(ds = df$ds,
holidays = rowSums(df[, holiday.comps]), holidays = rowSums(df[, holiday.comps, drop = FALSE]),
holidays_lower = rowSums(df[, paste0(holiday.comps, holidays_lower = rowSums(df[, paste0(holiday.comps,
"_lower")]), "_lower"), drop = FALSE]),
holidays_upper = rowSums(df[, paste0(holiday.comps, holidays_upper = rowSums(df[, paste0(holiday.comps,
"_upper")])) "_upper"), drop = FALSE]))
# NOTE the above CI calculation is incorrect if holidays overlap in time. # NOTE the above CI calculation is incorrect if holidays overlap in time.
# Since it is just for the visualization we will not worry about it now. # Since it is just for the visualization we will not worry about it now.
gg.holidays <- ggplot2::ggplot(df.s, ggplot2::aes(x = ds, y = holidays)) + gg.holidays <- ggplot2::ggplot(df.s, ggplot2::aes(x = ds, y = holidays)) +