From 05effe55ac48f0a3c56583f4d88e6496ddc1fe2d Mon Sep 17 00:00:00 2001 From: Maxim Grishin Date: Sun, 29 Sep 2019 23:14:40 +0300 Subject: [PATCH] Make StanFit4Model object accessible from Prophet --- python/fbprophet/forecaster.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/fbprophet/forecaster.py b/python/fbprophet/forecaster.py index c3f8a79..3635581 100644 --- a/python/fbprophet/forecaster.py +++ b/python/fbprophet/forecaster.py @@ -1116,9 +1116,9 @@ class Prophet(object): iter=self.mcmc_samples, ) args.update(kwargs) - stan_fit = model.sampling(**args) - for par in stan_fit.model_pars: - self.params[par] = stan_fit[par] + self.stan_fit = model.sampling(**args) + for par in self.stan_fit.model_pars: + self.params[par] = self.stan_fit[par] # Shape vector parameters if par in ['delta', 'beta'] and len(self.params[par].shape) < 2: self.params[par] = self.params[par].reshape((-1, 1))