From 48f5c7fa154aeada94052684127959a3f82986ab Mon Sep 17 00:00:00 2001 From: Bill Date: Tue, 23 Aug 2016 21:40:00 -0500 Subject: [PATCH] trying to make gp_kronecker models savable --- GPy/models/gp_kronecker_gaussian_regression.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/GPy/models/gp_kronecker_gaussian_regression.py b/GPy/models/gp_kronecker_gaussian_regression.py index 37a33bef..1d1b78f2 100644 --- a/GPy/models/gp_kronecker_gaussian_regression.py +++ b/GPy/models/gp_kronecker_gaussian_regression.py @@ -28,8 +28,14 @@ class GPKroneckerGaussianRegression(Model): } """ - def __init__(self, X1, X2, Y, kern1, kern2, noise_var=1., name='KGPR'): + def __init__(self, X1, X2, Y, kern1=None, kern2=None, noise_var=1., name='KGPR'): Model.__init__(self, name=name) + + if kern1 is None: + kern1 = GPy.kern.RBF(X1.shape[1]) + if kern2 is None: + kern2 = GPy.kern.RBF(X2.shape[1]) + # accept the construction arguments self.X1 = ObsAr(X1) self.X2 = ObsAr(X2)