Spaces:
Sleeping
Sleeping
MilesCranmer
commited on
Commit
•
9b325dd
1
Parent(s):
14e9a4b
Fix model_selection in test
Browse files- 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,
|