diff --git a/src/test_model.py b/src/test_model.py index 512c8ba..1dab024 100644 --- a/src/test_model.py +++ b/src/test_model.py @@ -47,7 +47,6 @@ def test_polynomial_regression_fitting(self): y_pred = model.transform(self.x) self.assertEqual(y_pred.shape, self.y.shape) self.assertAlmostEqual(y_pred.mean(), self.y.mean(), places=1) - print(model.weight) self.assertAlmostEqual(model.weight[0].item(), 0, places=0) self.assertAlmostEqual(model.weight[1].item(), 2, places=0) self.assertAlmostEqual(model.weight[2].item(), 5, places=0)