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

robustified the assertion re lengthscales in rbf

Browse source
This commit is contained in:
James Hensman 2013-01-31 09:02:18 +00:00
parent 3ad245eeec
commit 6197c1a8ed
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
GPy/kern/rbf.py
View file

@ -36,13 +36,15 @@ class rbf(kernpart):
if not ARD: if not ARD:
self.Nparam = 2 self.Nparam = 2
if lengthscale is not None: if lengthscale is not None:
assert lengthscale.size == 1 lengthscale = np.asarray(lengthscale)
assert lengthscale.size == 1, "Only one lengthscale needed for non-ARD kernel"
else: else:
lengthscale = np.ones(1) lengthscale = np.ones(1)
else: else:
self.Nparam = self.D + 1 self.Nparam = self.D + 1
if lengthscale is not None: if lengthscale is not None:
assert lengthscale.size == self.D lengthscale = np.asarray(lengthscale)
assert lengthscale.size == self.D, "bad number of lengthscales"
else: else:
lengthscale = np.ones(self.D) lengthscale = np.ones(self.D)