Spaces:
Running
Running
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,
|