MilesCranmer commited on
Commit
9b325dd
1 Parent(s): 14e9a4b

Fix model_selection in test

Browse files
Files changed (1) hide show
  1. test/test.py +0 -2
test/test.py CHANGED
@@ -27,7 +27,6 @@ class TestPipeline(unittest.TestCase):
27
  y = self.X[:, 0]
28
  model = PySRRegressor(**self.default_test_kwargs)
29
  model.fit(self.X, y)
30
- model.set_params(model_selection="accuracy")
31
  print(model.equations)
32
  self.assertLessEqual(model.get_best()["loss"], 1e-4)
33
 
@@ -100,7 +99,6 @@ class TestPipeline(unittest.TestCase):
100
  X = np.random.randn(100, 1)
101
  y = X[:, 0] + 3.0
102
  regressor = PySRRegressor(
103
- model_selection="accuracy",
104
  unary_operators=[],
105
  binary_operators=["plus"],
106
  **self.default_test_kwargs,
 
27
  y = self.X[:, 0]
28
  model = PySRRegressor(**self.default_test_kwargs)
29
  model.fit(self.X, y)
 
30
  print(model.equations)
31
  self.assertLessEqual(model.get_best()["loss"], 1e-4)
32
 
 
99
  X = np.random.randn(100, 1)
100
  y = X[:, 0] + 3.0
101
  regressor = PySRRegressor(
 
102
  unary_operators=[],
103
  binary_operators=["plus"],
104
  **self.default_test_kwargs,