From de5b7494de00c3d486ade98679f2fbb5f587be08 Mon Sep 17 00:00:00 2001 From: Ben Letham Date: Tue, 15 Jan 2019 12:47:54 -0800 Subject: [PATCH] Remove pd.Series from np.matmul --- python/fbprophet/forecaster.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/fbprophet/forecaster.py b/python/fbprophet/forecaster.py index e800f38..bfe8f40 100644 --- a/python/fbprophet/forecaster.py +++ b/python/fbprophet/forecaster.py @@ -1369,8 +1369,8 @@ class Prophet(object): trend = self.sample_predictive_trend(df, iteration) beta = self.params['beta'][iteration] - Xb_a = np.matmul(seasonal_features.values, beta * s_a) * self.y_scale - Xb_m = np.matmul(seasonal_features.values, beta * s_m) + Xb_a = np.matmul(seasonal_features.values, beta * s_a.values) * self.y_scale + Xb_m = np.matmul(seasonal_features.values, beta * s_m.values) sigma = self.params['sigma_obs'][iteration] noise = np.random.normal(0, sigma, df.shape[0]) * self.y_scale