Browse Source

Fit if constant history and logistic growth

bl 7 years ago
parent
commit
1b91fa3064
1 changed files with 4 additions and 1 deletions
  1. 4 1
      python/fbprophet/forecaster.py

+ 4 - 1
python/fbprophet/forecaster.py

@@ -980,7 +980,10 @@ class Prophet(object):
                 'sigma_obs': 1,
             }
 
-        if history['y'].min() == history['y'].max():
+        if (
+            (history['y'].min() == history['y'].max())
+            and self.growth == 'linear'
+        ):
             # Nothing to fit.
             self.params = stan_init()
             self.params['sigma_obs'] = 1e-9