def test_predict(self):
'''
Test the predict function.
'''
for _ in range(ITERATIONS):
indim = random_size()
outdim = random_size()
coeff = np.random.randn(indim, outdim)
model = UpdatingModel(indim, outdim)
inp = np.random.randn(SAMPLES, indim)
out = coeff.T.dot(inp.T).T
model.add_samples(inp, out)
pred = model.predict(inp[0, :])
self.assertTrue(sum(abs(pred - out[0, :])) < DIFF)
def test_add_samples(self):
'''
Tests the add_samples method.
'''
for _ in range(ITERATIONS):
indim = random_size()
outdim = random_size()
coeff = np.random.randn(indim, outdim)
model = UpdatingModel(indim, outdim)
inp = np.random.randn(SAMPLES, indim)
out = coeff.T.dot(inp.T).T
model.add_samples(inp, out)
diff = sum(sum(abs(model.coeff.T - coeff)))
self.assertTrue(diff < DIFF)
