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

Use explicity cython/numpy variants in coregionalize test

Browse source
This commit is contained in:
Jayanth Koushik 2018-02-15 17:18:03 -05:00
parent 1aab041885
commit 64c125573e
1 changed files with 4 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

18
GPy/testing/kernel_tests.py
View file

@ -631,32 +631,25 @@ class Coregionalize_cython_test(unittest.TestCase):
def test_sym(self): def test_sym(self):
dL_dK = np.random.randn(self.N1, self.N1) dL_dK = np.random.randn(self.N1, self.N1)
GPy.util.config.config.set('cython', 'working', 'True') K_cython = self.k._K_cython(self.X)
K_cython = self.k.K(self.X)
self.k.update_gradients_full(dL_dK, self.X) self.k.update_gradients_full(dL_dK, self.X)
grads_cython = self.k.gradient.copy() grads_cython = self.k.gradient.copy()
GPy.util.config.config.set('cython', 'working', 'False') K_numpy = self.k._K_numpy(self.X)
K_numpy = self.k.K(self.X)
self.k.update_gradients_full(dL_dK, self.X) self.k.update_gradients_full(dL_dK, self.X)
grads_numpy = self.k.gradient.copy() grads_numpy = self.k.gradient.copy()
self.assertTrue(np.allclose(K_numpy, K_cython)) self.assertTrue(np.allclose(K_numpy, K_cython))
self.assertTrue(np.allclose(grads_numpy, grads_cython)) self.assertTrue(np.allclose(grads_numpy, grads_cython))
#reset the cython state for any other tests
GPy.util.config.config.set('cython', 'working', 'true')
def test_nonsym(self): def test_nonsym(self):
dL_dK = np.random.randn(self.N1, self.N2) dL_dK = np.random.randn(self.N1, self.N2)
GPy.util.config.config.set('cython', 'working', 'True') K_cython = self.k._K_cython(self.X, self.X2)
K_cython = self.k.K(self.X, self.X2)
self.k.gradient = 0. self.k.gradient = 0.
self.k.update_gradients_full(dL_dK, self.X, self.X2) self.k.update_gradients_full(dL_dK, self.X, self.X2)
grads_cython = self.k.gradient.copy() grads_cython = self.k.gradient.copy()
GPy.util.config.config.set('cython', 'working', 'False') K_numpy = self.k._K_numpy(self.X, self.X2)
K_numpy = self.k.K(self.X, self.X2)
self.k.gradient = 0. self.k.gradient = 0.
self.k.update_gradients_full(dL_dK, self.X, self.X2) self.k.update_gradients_full(dL_dK, self.X, self.X2)
grads_numpy = self.k.gradient.copy() grads_numpy = self.k.gradient.copy()
@ -664,9 +657,6 @@ class Coregionalize_cython_test(unittest.TestCase):
self.assertTrue(np.allclose(K_numpy, K_cython)) self.assertTrue(np.allclose(K_numpy, K_cython))
self.assertTrue(np.allclose(grads_numpy, grads_cython)) self.assertTrue(np.allclose(grads_numpy, grads_cython))
#reset the cython state for any other tests
GPy.util.config.config.set('cython', 'working', 'true')
class KernelTestsProductWithZeroValues(unittest.TestCase): class KernelTestsProductWithZeroValues(unittest.TestCase):