Просмотр исходного кода

Fall back to Newton optimizer if LBFGS fails

bl 8 лет назад
Родитель
Сommit
f7becb0942
1 измененных файлов с 8 добавлено и 1 удалено
  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))