Spaces:
Running
Running
MilesCranmer
commited on
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)
|