Explorar o código

Fall back to Newton optimizer if LBFGS fails

bl %!s(int64=8) %!d(string=hai) anos
pai
achega
f7becb0942
Modificáronse 1 ficheiros con 8 adicións e 1 borrados
  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))