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

optionally unweaved the coregionalize kernel

Browse source 
coregionalize shoudl now work without weave. Added kernel tests also.
This commit is contained in:
James Hensman 2014-09-29 10:09:28 +01:00
parent afbb8ab253
commit 9081c8ee96
2 changed files with 99 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

44
GPy/testing/kernel_tests.py
View file

@ -356,6 +356,50 @@ class KernelTestsNonContinuous(unittest.TestCase):
X2 = self.X2[self.X2[:,-1]!=2]
self.assertTrue(check_kernel_gradient_functions(kern, X=X, X2=X2, verbose=verbose, fixed_X_dims=-1))
class Coregionalize_weave_test(unittest.TestCase):
"""
Make sure that the coregionalize kernel work with and without weave enabled
"""
k = GPy.kern.coregionalize(1, output_dim=12)
N1, N2 = 100, 200
X = np.random.randint(0,12,(N1,1))
X2 = np.random.randint(0,12,(N2,1))
#symmetric case
dL_dK = np.random.randn(N1, N1)
GPy.util.config.config.set('weave', 'working', True)
K_weave = k.K(X)
k.update_gradients_full(dL_dK, X)
grads_weave = k.gradient.copy()
GPy.util.config.config.set('weave', 'working', False)
K_numpy = k.K(X)
k.update_gradients_full(dL_dK, X)
grads_numpy = k.gradient.copy()
self.assertTrue(np.allclose(K_numpy, K_weave))
self.assertTrue(np.allclose(grads_numpy, grads_weave))
#non-symmetric case
dL_dK = np.random.randn(N1, N2)
GPy.util.config.config.set('weave', 'working', True)
K_weave = k.K(X, X2)
k.update_gradients_full(dL_dK, X, X2)
grads_weave = k.gradient.copy()
GPy.util.config.config.set('weave', 'working', False)
K_numpy = k.K(X, X2)
k.update_gradients_full(dL_dK, X, X2)
grads_numpy = k.gradient.copy()
self.assertTrue(np.allclose(K_numpy, K_weave))
self.assertTrue(np.allclose(grads_numpy, grads_weave))
#reset the weave state for any other tests
GPy.util.config.config.set('weave', 'working', False)
if __name__ == "__main__":
print "Running unit tests, please be (very) patient..."