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

Fix usage of uncertainty.samples

Browse source
This commit is contained in:
Mitchell 2019-10-09 15:50:18 +11:00 committed by Ben Letham
parent a2a0df904e
commit f4d997e0ac
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
R/R/prophet.R
View file

@ -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(