From 42a415e29946080b29f4f798f043161236183a9a Mon Sep 17 00:00:00 2001 From: Alan Saul Date: Wed, 22 Jul 2015 18:37:12 +0100 Subject: [PATCH] Fixed model test --- GPy/models/gp_var_gauss.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/GPy/models/gp_var_gauss.py b/GPy/models/gp_var_gauss.py index ccaab305..729b6bb8 100644 --- a/GPy/models/gp_var_gauss.py +++ b/GPy/models/gp_var_gauss.py @@ -9,6 +9,7 @@ from ..core.parameterization import ObsAr from .. import kern from ..core.parameterization.param import Param from ..util.linalg import pdinv +from ..likelihoods import Gaussian log_2_pi = np.log(2*np.pi) @@ -25,8 +26,10 @@ class GPVariationalGaussianApproximation(Model): pages = {786--792}, } """ - def __init__(self, X, Y, kernel, likelihood,Y_metadata=None): - Model.__init__(self,'Variational GP classification') + def __init__(self, X, Y, kernel, likelihood=None, Y_metadata=None): + Model.__init__(self,'Variational GP') + if likelihood is None: + likelihood = Gaussian() # accept the construction arguments self.X = ObsAr(X) self.Y = Y