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

start of cythoning coregionalize

Browse source
This commit is contained in:
James Hensman 2015-04-27 16:00:42 +01:00
parent 2e8ce34ee0
commit c00f76d250
3 changed files with 56 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -111,6 +111,11 @@ class Coregionalize(Kern):
weave.inline(code, ['target', 'index', 'index2', 'N', 'num_inducing', 'B', 'output_dim'])
return target
def _K_cython(self, X, X2=None):
if X2 is None:
return coregionalize_cython.K_symmetric(self.B, X[:,0])
return coregionalize_cython.K_asymmetric(self.B, X[:,0], X2[:,0])
def Kdiag(self, X):
return np.diag(self.B)[np.asarray(X, dtype=np.int).flatten()]
@ -164,6 +169,11 @@ class Coregionalize(Kern):
dL_dK_small[j,i] = tmp1[:,index2==j].sum()
return dL_dK_small
def gradient_reduce_cython(self, dL_dK, index, index2):
index, index2 = index[:,0], index2[:,0]
return coregionalize_cython.gradient_reduce(self.output_dim, dL_dK, index, index2
def update_gradients_diag(self, dL_dKdiag, X):
index = np.asarray(X, dtype=np.int).flatten()
dL_dKdiag_small = np.array([dL_dKdiag[index==i].sum() for i in range(self.output_dim)])