fixed coregionalize cython

This commit is contained in:
James Hensman 2015-04-28 15:57:27 +01:00
parent 9e4c33910f
commit 1ebbc49fdd
5 changed files with 437 additions and 232 deletions

View file

@ -73,8 +73,8 @@ class Coregionalize(Kern):
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])
return coregionalize_cython.K_symmetric(self.B, np.asarray(X, dtype=np.int64)[:,0])
return coregionalize_cython.K_asymmetric(self.B, np.asarray(X, dtype=np.int64)[:,0], np.asarray(X2, dtype=np.int64)[:,0])
def Kdiag(self, X):