var[:,None] added in full_cov = false, sparse_GP

This commit is contained in:
Ricardo Andrade 2013-02-04 12:01:27 +00:00
parent 7a54660689
commit 049e60f16b
2 changed files with 3 additions and 4 deletions

View file

@ -196,15 +196,14 @@ class sparse_GP(GP):
Kx = self.kern.K(self.Z, Xnew)
mu = mdot(Kx.T, self.C/self.scale_factor, self.psi1V)
if full_cov:
Kxx = self.kern.K(Xnew)
var = Kxx - mdot(Kx.T, (self.Kmmi - self.C/self.scale_factor**2), Kx)
var = Kxx - mdot(Kx.T, (self.Kmmi - self.C/self.scale_factor**2), Kx) #NOTE thiswon't work for plotting
else:
Kxx = self.kern.Kdiag(Xnew)
var = Kxx - np.sum(Kx*np.dot(self.Kmmi - self.C/self.scale_factor**2, Kx),0)
return mu,var
return mu,var[:,None]
def plot(self, *args, **kwargs):
"""