From 671ae41256574a99211917cefc98084f2cfd5d95 Mon Sep 17 00:00:00 2001 From: Nicolo Fusi Date: Mon, 15 Apr 2013 16:02:02 +0100 Subject: [PATCH] small changes --- GPy/inference/SGD.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/GPy/inference/SGD.py b/GPy/inference/SGD.py index 5d1b673d..be1c92de 100644 --- a/GPy/inference/SGD.py +++ b/GPy/inference/SGD.py @@ -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