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

bug fix

Browse source
This commit is contained in:
Zhenwen Dai 2014-10-24 09:11:03 +01:00
parent bc3dab4c4d
commit db0a94e1a3
2 changed files with 15 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
GPy/inference/latent_function_inference/inference_X.py
View file

@ -26,6 +26,10 @@ class Inference_X(Model):
self.likelihood = model.likelihood.copy() self.likelihood = model.likelihood.copy()
self.kern = model.kern.copy() self.kern = model.kern.copy()
if model.kern.useGPU: if model.kern.useGPU:
from ...models import SSGPLVM
if isinstance(model, SSGPLVM):
self.kern.GPU_SSRBF(True)
else:
self.kern.GPU(True) self.kern.GPU(True)
from copy import deepcopy from copy import deepcopy
self.posterior = deepcopy(model.posterior) self.posterior = deepcopy(model.posterior)

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

@ -41,6 +41,16 @@ class RBF(Stationary):
if isinstance(self.psicomp,PSICOMP_RBF_GPU): if isinstance(self.psicomp,PSICOMP_RBF_GPU):
self.psicomp = PSICOMP_RBF() self.psicomp = PSICOMP_RBF()
def GPU_SSRBF(self,flag=True):
if flag:
if isinstance(self.psicomp,PSICOMP_RBF):
self.useGPU = True
from psi_comp.ssrbf_psi_gpucomp import PSICOMP_SSRBF_GPU
self.psicomp = PSICOMP_SSRBF_GPU()
else:
if isinstance(self.psicomp,PSICOMP_RBF_GPU):
self.psicomp = PSICOMP_RBF()
def K_of_r(self, r): def K_of_r(self, r):
return self.variance * np.exp(-0.5 * r**2) return self.variance * np.exp(-0.5 * r**2)