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

[GPU] psi1

Browse source
This commit is contained in:
Zhenwen Dai 2014-03-31 16:18:06 +01:00
parent 4465c5be8d
commit b945e8d01f
4 changed files with 114 additions and 109 deletions
Download patch file Download diff file Expand all files Collapse all files

4
GPy/kern/_src/rbf.py
View file

@ -21,8 +21,8 @@ class RBF(Stationary):
"""
_support_GPU = True
def __init__(self, input_dim, variance=1., lengthscale=None, ARD=False, active_dims=None, name='rbf'):
super(RBF, self).__init__(input_dim, variance, lengthscale, ARD, active_dims, name)
def __init__(self, input_dim, variance=1., lengthscale=None, ARD=False, active_dims=None, name='rbf', useGPU=False):
super(RBF, self).__init__(input_dim, variance, lengthscale, ARD, active_dims, name, useGPU=useGPU)
self.weave_options = {}
self.group_spike_prob = False