apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-15 06:52:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

[opt] unified printing of status of optimization

Browse source
This commit is contained in:
Max Zwiessele 2015-01-15 09:02:45 +00:00
parent 4deac0103c
commit 57e941140c
4 changed files with 17 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

3
GPy/core/model.py
View file

@ -256,8 +256,9 @@ class Model(Parameterized):
optimizer = optimization.get_optimizer(optimizer)
opt = optimizer(start, model=self, max_iters=max_iters, **kwargs)
with VerboseOptimization(self, opt, maxiters=max_iters, verbose=messages, ipython_notebook=ipython_notebook):
with VerboseOptimization(self, opt, maxiters=max_iters, verbose=messages, ipython_notebook=ipython_notebook) as vo:
opt.run(f_fp=self._objective_grads, f=self._objective, fp=self._grads)
vo.finish(opt)
self.optimization_runs.append(opt)