if optimization fails fallback to Newton even if algorithm is specified (#890)

This commit is contained in:
MT 2019-03-18 09:41:45 -07:00 committed by Ben Letham
parent 730e59e2f7
commit eec2f46ac5

View file

@ -1096,12 +1096,10 @@ class Prophet(object):
try:
params = model.optimizing(**args)
except RuntimeError:
if 'algorithm' not in args:
# Fall back on Newton
args['algorithm'] = 'Newton'
params = model.optimizing(**args)
else:
raise
# Fall back on Newton
args['algorithm'] = 'Newton'
params = model.optimizing(**args)
for par in params:
self.params[par] = params[par].reshape((1, -1))