normalize_Y is passed to core function

This commit is contained in:
Ricardo 2013-09-17 17:44:00 +01:00
parent a1a8486bc3
commit 2762e158fd

View file

@ -33,7 +33,7 @@ class GPLVM(GP):
if kernel is None: if kernel is None:
kernel = kern.rbf(input_dim, ARD=input_dim > 1) + kern.bias(input_dim, np.exp(-2)) kernel = kern.rbf(input_dim, ARD=input_dim > 1) + kern.bias(input_dim, np.exp(-2))
likelihood = Gaussian(Y, normalize=normalize_Y, variance=np.exp(-2.)) likelihood = Gaussian(Y, normalize=normalize_Y, variance=np.exp(-2.))
GP.__init__(self, X, likelihood, kernel, normalize_X=False) GP.__init__(self, X, likelihood, kernel, normalize_X=False,normalize_Y = normalize_Y)
self.set_prior('.*X', Gaussian_prior(0, 1)) self.set_prior('.*X', Gaussian_prior(0, 1))
self.ensure_default_constraints() self.ensure_default_constraints()