apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-07 19:12:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

[model] optimizer can now be an optimizer instance, instead of a string

Browse source
This commit is contained in:
mzwiessele 2014-06-19 12:06:39 -07:00
parent 1e006f63b5
commit eaa18e3512
1 changed files with 12 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

8
GPy/core/model.py
View file

@ -227,13 +227,15 @@ class Model(Parameterized):
if optimizer is None: if optimizer is None:
optimizer = self.preferred_optimizer optimizer = self.preferred_optimizer
elif isinstance(optimizer, optimization.Optimizer):
opt = optimizer
else:
optimizer = optimization.get_optimizer(optimizer)
opt = optimizer(start, model=self, **kwargs)
if start == None: if start == None:
start = self.optimizer_array start = self.optimizer_array
optimizer = optimization.get_optimizer(optimizer)
opt = optimizer(start, model=self, **kwargs)
opt.run(f_fp=self._objective_grads, f=self._objective, fp=self._grads) opt.run(f_fp=self._objective_grads, f=self._objective, fp=self._grads)
self.optimization_runs.append(opt) self.optimization_runs.append(opt)