apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-10 20:42:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

trying to make gp_kronecker models savable

Browse source
This commit is contained in:
Bill 2016-08-23 21:40:00 -05:00
parent 5cb22090ba
commit 48f5c7fa15
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
GPy/models/gp_kronecker_gaussian_regression.py
View file

@ -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)