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)