apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-07-11 16:22:13 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update static.py

Browse source
This commit is contained in:
Joan Gonzalvez 2019-01-13 20:16:32 +01:00 • committed by GitHub
parent 400a4880c6
commit b41b0df9a0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
GPy/kern/src/static.py
View file

@ -64,6 +64,11 @@ class White(Static):
def __init__(self, input_dim, variance=1., active_dims=None, name='white'): def __init__(self, input_dim, variance=1., active_dims=None, name='white'):
super(White, self).__init__(input_dim, variance, active_dims, name) super(White, self).__init__(input_dim, variance, active_dims, name)
def to_dict(self):
input_dict = super(White, self)._save_to_input_dict()
input_dict["class"] = "GPy.kern.White"
return input_dict
def K(self, X, X2=None): def K(self, X, X2=None):
if X2 is None: if X2 is None:
return np.eye(X.shape[0])*self.variance return np.eye(X.shape[0])*self.variance
@ -102,6 +107,10 @@ class WhiteHeteroscedastic(Static):
super(Static, self).__init__(input_dim, active_dims, name) super(Static, self).__init__(input_dim, active_dims, name)
self.variance = Param('variance', np.ones(num_data) * variance, Logexp()) self.variance = Param('variance', np.ones(num_data) * variance, Logexp())
self.link_parameters(self.variance) self.link_parameters(self.variance)
def to_dict(self):
input_dict = super(WhiteHeteroscedastic, self)._save_to_input_dict()
input_dict["class"] = "GPy.kern.WhiteHeteroscedastic"
return input_dict
def Kdiag(self, X): def Kdiag(self, X):
if X.shape[0] == self.variance.shape[0]: if X.shape[0] == self.variance.shape[0]: