浏览代码

Fall back to Newton optimizer if LBFGS fails

bl 8 年之前
父节点
当前提交
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))