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

small changes

Browse source
This commit is contained in:
Nicolo Fusi 2013-04-15 16:02:02 +01:00
parent df7cfa62a7
commit 671ae41256
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
GPy/inference/SGD.py
View file

@ -230,6 +230,8 @@ class opt_SGD(Optimizer):
shapes = self.get_param_shapes(N, Q)
f, step, Nj = self.step_with_missing_data(f_fp, X, step, shapes)
else:
self.model.likelihood.YYT = np.dot(self.model.likelihood.Y, self.model.likelihood.Y.T)
self.model.likelihood.trYYT = np.trace(self.model.likelihood.YYT)
Nj = N
f, fp = f_fp(self.x_opt)
step = self.momentum * step + self.learning_rate * fp