Changes to MRD test

This commit is contained in:
Ricardo 2013-05-20 16:59:35 +01:00
parent 255a9bbd73
commit f01abb849f
2 changed files with 4 additions and 1 deletions

View file

@ -32,6 +32,8 @@ class Bayesian_GPLVM(sparse_GP, GPLVM):
**kwargs): **kwargs):
if type(likelihood_or_Y) is np.ndarray: if type(likelihood_or_Y) is np.ndarray:
likelihood = Gaussian(likelihood_or_Y) likelihood = Gaussian(likelihood_or_Y)
else:
likelihood = likelihood_or_Y
if X == None: if X == None:
X = self.initialise_latent(init, Q, likelihood.Y) X = self.initialise_latent(init, Q, likelihood.Y)

View file

@ -21,8 +21,9 @@ class MRDTests(unittest.TestCase):
K = k.K(X) K = k.K(X)
Ylist = [np.random.multivariate_normal(np.zeros(N), K, D).T for _ in range(num_m)] Ylist = [np.random.multivariate_normal(np.zeros(N), K, D).T for _ in range(num_m)]
likelihood_list = [GPy.likelihoods.Gaussian(Y) for Y in Ylist]
m = GPy.models.MRD(*Ylist, Q=Q, kernel=k, M=M) m = GPy.models.MRD(*likelihood_list, Q=Q, kernel=k, M=M)
m.ensure_default_constraints() m.ensure_default_constraints()
self.assertTrue(m.checkgrad()) self.assertTrue(m.checkgrad())