More ARD flags in exponential and Matern32

This commit is contained in:
Nicolas 2013-01-18 15:34:06 +00:00
parent 11d088cf90
commit 0c30048e57
2 changed files with 40 additions and 25 deletions

View file

@ -49,13 +49,13 @@ class exponential(kernpart):
def _set_params(self,x):
"""set the value of the parameters."""
assert x.size==(self.D+1)
assert x.size == self.Nparam
self.variance = x[0]
self.lengthscale = x[1:]
def _get_param_names(self):
"""return parameter names."""
if self.Nparam==2:
if self.Nparam == 2:
return ['variance','lengthscale']
else:
return ['variance']+['lengthscale_%i'%i for i in range(self.lengthscale.size)]
@ -84,7 +84,6 @@ class exponential(kernpart):
else:
dl = self.variance*dvar*dist2M.sum(-1)*invdist
target[1] += np.sum(dl*partial)
#foo
def dKdiag_dtheta(self,partial,X,target):
"""derivative of the diagonal of the covariance matrix with respect to the parameters."""