Parcourir la source

Fix bug in seasonal components with MCMC

Ben Letham il y a 7 ans
Parent
commit
bc6360d115
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2 2
      R/R/prophet.R

+ 2 - 2
R/R/prophet.R

@@ -1304,9 +1304,9 @@ predict_seasonal_components <- function(m, df) {
   X <- as.matrix(seasonal.features)
   component.predictions <- data.frame(matrix(ncol = 0, nrow = nrow(X)))
   for (component in colnames(component.cols)) {
-    beta.c <- m$params$beta * component.cols[[component]]
+    beta.c <- t(m$params$beta) * component.cols[[component]]
 
-    comp <- X %*% t(beta.c)
+    comp <- X %*% beta.c
     if (component %in% m$component.modes$additive) {
       comp <- comp * m$y.scale
     }