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

[sparse gp] prediction with posterior per dimension activated

Browse source
This commit is contained in:
Max Zwiessele 2014-10-29 08:23:25 +00:00
parent f91e7e9deb
commit af40ef8cfb
2 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
GPy/likelihoods/gaussian.py
View file

@ -80,7 +80,10 @@ class Gaussian(Likelihood):
def predictive_values(self, mu, var, full_cov=False, Y_metadata=None):
if full_cov:
var += np.eye(var.shape[0])*self.variance
if var.ndim == 2:
var += np.eye(var.shape[0])*self.variance
if var.ndim == 3:
var += np.atleast_3d(np.eye(var.shape[0])*self.variance)
else:
var += self.variance
return mu, var