From f4d997e0acb286507182cb9f5aefeaaa17ff675b Mon Sep 17 00:00:00 2001 From: Mitchell Date: Wed, 9 Oct 2019 15:50:18 +1100 Subject: [PATCH] Fix usage of uncertainty.samples --- R/R/prophet.R | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/R/R/prophet.R b/R/R/prophet.R index 484a1cc..fe851f4 100644 --- a/R/R/prophet.R +++ b/R/R/prophet.R @@ -1343,7 +1343,7 @@ predict.prophet <- function(object, df = NULL, ...) { df$trend <- predict_trend(object, df) seasonal.components <- predict_seasonal_components(object, df) - if (uncertainty.samples) { + if (object$uncertainty.samples) { intervals <- predict_uncertainty(object, df) } else { intervals <- NULL @@ -1458,7 +1458,7 @@ predict_seasonal_components <- function(m, df) { m <- out$m seasonal.features <- out$seasonal.features component.cols <- out$component.cols - if (uncertainty.samples){ + if (m$uncertainty.samples){ lower.p <- (1 - m$interval.width)/2 upper.p <- (1 + m$interval.width)/2 } @@ -1473,7 +1473,7 @@ predict_seasonal_components <- function(m, df) { comp <- comp * m$y.scale } component.predictions[[component]] <- rowMeans(comp, na.rm = TRUE) - if (uncertainty.samples){ + if (m$uncertainty.samples){ component.predictions[[paste0(component, '_lower')]] <- apply( comp, 1, stats::quantile, lower.p, na.rm = TRUE) component.predictions[[paste0(component, '_upper')]] <- apply(