Ver código fonte

Fall back to Newton optimizer if LBFGS fails

bl 8 anos atrás
pai
commit
f7becb0942
1 arquivos alterados com 8 adições e 1 exclusões
  1. 8 1
      python/fbprophet/forecaster.py

+ 8 - 1
python/fbprophet/forecaster.py

@@ -539,7 +539,14 @@ class Prophet(object):
                 self.params[par] = stan_fit[par]
 
         else:
-            params = model.optimizing(dat, init=stan_init, iter=1e4, **kwargs)
+            try:
+                params = model.optimizing(
+                    dat, init=stan_init, iter=1e4, **kwargs)
+            except RuntimeError:
+                params = model.optimizing(
+                    dat, init=stan_init, iter=1e4, algorithm='Newton',
+                    **kwargs
+                )
             for par in params:
                 self.params[par] = params[par].reshape((1, -1))