small mod in toy_ARD

This commit is contained in:
Andreas 2013-07-15 19:23:48 +01:00
parent e798eaa34b
commit edfd6ea948

View file

@ -78,6 +78,8 @@ def toy_ARD(optim_iters=1000, kernel_type='linear', N=300, D=4):
if kernel_type == 'linear': if kernel_type == 'linear':
kernel = GPy.kern.linear(X.shape[1], ARD = 1) kernel = GPy.kern.linear(X.shape[1], ARD = 1)
elif kernel_type == 'rbf_inv':
kernel = GPy.kern.rbf_inv(X.shape[1], ARD = 1)
else: else:
kernel = GPy.kern.rbf(X.shape[1], ARD = 1) kernel = GPy.kern.rbf(X.shape[1], ARD = 1)
kernel += GPy.kern.white(X.shape[1]) + GPy.kern.bias(X.shape[1]) kernel += GPy.kern.white(X.shape[1]) + GPy.kern.bias(X.shape[1])