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

Update self.num_data in GP when X is updated

Browse source
This commit is contained in:
Masha Naslidnyk 🦉 2019-09-05 14:39:13 +01:00
parent 40137cc8f7
commit 1e06c6ce2f
2 changed files with 16 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
GPy/core/gp.py
View file

@ -235,6 +235,8 @@ class GP(Model):
self.link_parameter(self.X, index=index) self.link_parameter(self.X, index=index)
else: else:
self.X = ObsAr(X) self.X = ObsAr(X)
self.num_data, self.input_dim = self.X.shape
self.update_model(True) self.update_model(True)
def set_X(self,X): def set_X(self,X):

13
GPy/testing/gp_tests.py
View file

@ -28,12 +28,15 @@ class Test(unittest.TestCase):
Xnew = NormalPosterior(m.X.mean[:10].copy(), m.X.variance[:10].copy()) Xnew = NormalPosterior(m.X.mean[:10].copy(), m.X.variance[:10].copy())
m.set_XY(Xnew, m.Y[:10].copy()) m.set_XY(Xnew, m.Y[:10].copy())
assert(m.checkgrad()) assert(m.checkgrad())
assert(m.num_data == m.X.shape[0])
assert(m.input_dim == m.X.shape[1])
m.set_XY(X, self.Y) m.set_XY(X, self.Y)
mu2, var2 = m.predict(m.X) mu2, var2 = m.predict(m.X)
np.testing.assert_allclose(mu, mu2) np.testing.assert_allclose(mu, mu2)
np.testing.assert_allclose(var, var2) np.testing.assert_allclose(var, var2)
def test_setxy_gplvm(self): def test_setxy_gplvm(self):
k = GPy.kern.RBF(1) k = GPy.kern.RBF(1)
m = GPy.models.GPLVM(self.Y, 1, kernel=k) m = GPy.models.GPLVM(self.Y, 1, kernel=k)
@ -42,6 +45,10 @@ class Test(unittest.TestCase):
Xnew = X[:10].copy() Xnew = X[:10].copy()
m.set_XY(Xnew, m.Y[:10].copy()) m.set_XY(Xnew, m.Y[:10].copy())
assert(m.checkgrad()) assert(m.checkgrad())
assert(m.num_data == m.X.shape[0])
assert(m.input_dim == m.X.shape[1])
m.set_XY(X, self.Y) m.set_XY(X, self.Y)
mu2, var2 = m.predict(m.X) mu2, var2 = m.predict(m.X)
np.testing.assert_allclose(mu, mu2) np.testing.assert_allclose(mu, mu2)
@ -54,6 +61,10 @@ class Test(unittest.TestCase):
X = m.X.copy() X = m.X.copy()
m.set_XY(m.X[:10], m.Y[:10]) m.set_XY(m.X[:10], m.Y[:10])
assert(m.checkgrad()) assert(m.checkgrad())
assert(m.num_data == m.X.shape[0])
assert(m.input_dim == m.X.shape[1])
m.set_XY(X, self.Y) m.set_XY(X, self.Y)
mu2, var2 = m.predict(m.X) mu2, var2 = m.predict(m.X)
np.testing.assert_allclose(mu, mu2) np.testing.assert_allclose(mu, mu2)