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

fixed bug in rbf.py, removed the ARD moniker from the name variable

Browse source
This commit is contained in:
James Hensman 2013-01-31 08:54:57 +00:00
parent 5176f4c6f3
commit b7462e0fcc
1 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
GPy/kern/rbf.py
View file

@ -26,28 +26,27 @@ class rbf(kernpart):
:type ARD: Boolean :type ARD: Boolean
:rtype: kernel object :rtype: kernel object
.. Note: for rbf with different lengthscale on each dimension, see rbf_ARD .. Note: this object implements both the ARD and 'spherical' version of the function
""" """
def __init__(self,D,variance=1.,lengthscale=None,ARD=False): def __init__(self,D,variance=1.,lengthscale=None,ARD=False):
self.D = D self.D = D
self.name = 'rbf'
self.ARD = ARD self.ARD = ARD
if not ARD: if not ARD:
self.Nparam = 2 self.Nparam = 2
self.name = 'rbf'
if lengthscale is not None: if lengthscale is not None:
assert lengthscale.shape == (1,) assert lengthscale.size == 1
else: else:
lengthscale = np.ones(1) lengthscale = np.ones(1)
else: else:
self.Nparam = self.D + 1 self.Nparam = self.D + 1
self.name = 'rbf_ARD'
if lengthscale is not None: if lengthscale is not None:
assert lengthscale.shape == (self.D,) assert lengthscale.size == self.D
else: else:
lengthscale = np.ones(self.D) lengthscale = np.ones(self.D)
self._set_params(np.hstack((variance,lengthscale))) self._set_params(np.hstack((variance,lengthscale.flatten())))
#initialize cache #initialize cache
self._Z, self._mu, self._S = np.empty(shape=(3,1)) self._Z, self._mu, self._S = np.empty(shape=(3,1))