Spaces:
Running
Running
Commit
·
98ccd5e
1
Parent(s):
c526fa1
Add `turbo` parameter with test
Browse files- pysr/sr.py +0 -1
- test/test.py +2 -1
pysr/sr.py
CHANGED
|
@@ -1587,7 +1587,6 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
|
|
| 1587 |
early_stop_condition=early_stop_condition,
|
| 1588 |
seed=seed,
|
| 1589 |
deterministic=self.deterministic,
|
| 1590 |
-
extend_user_operators=False,
|
| 1591 |
define_helper_functions=False,
|
| 1592 |
)
|
| 1593 |
|
|
|
|
| 1587 |
early_stop_condition=early_stop_condition,
|
| 1588 |
seed=seed,
|
| 1589 |
deterministic=self.deterministic,
|
|
|
|
| 1590 |
define_helper_functions=False,
|
| 1591 |
)
|
| 1592 |
|
test/test.py
CHANGED
|
@@ -70,12 +70,13 @@ class TestPipeline(unittest.TestCase):
|
|
| 70 |
print(model.equations_)
|
| 71 |
self.assertLessEqual(model.get_best()["loss"], 1e-4)
|
| 72 |
|
| 73 |
-
def
|
| 74 |
y = self.X[:, 0]
|
| 75 |
model = PySRRegressor(
|
| 76 |
**self.default_test_kwargs,
|
| 77 |
procs=2,
|
| 78 |
multithreading=False,
|
|
|
|
| 79 |
early_stop_condition="stop_if(loss, complexity) = loss < 1e-4 && complexity == 1",
|
| 80 |
)
|
| 81 |
model.fit(self.X, y)
|
|
|
|
| 70 |
print(model.equations_)
|
| 71 |
self.assertLessEqual(model.get_best()["loss"], 1e-4)
|
| 72 |
|
| 73 |
+
def test_multiprocessing_turbo(self):
|
| 74 |
y = self.X[:, 0]
|
| 75 |
model = PySRRegressor(
|
| 76 |
**self.default_test_kwargs,
|
| 77 |
procs=2,
|
| 78 |
multithreading=False,
|
| 79 |
+
turbo=True,
|
| 80 |
early_stop_condition="stop_if(loss, complexity) = loss < 1e-4 && complexity == 1",
|
| 81 |
)
|
| 82 |
model.fit(self.X, y)
|