Jelajahi Sumber

Fix bug in seasonal components with MCMC

Ben Letham 7 tahun lalu
induk
melakukan
bc6360d115
1 mengubah file dengan 2 tambahan dan 2 penghapusan
  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
     }