Spaces:
Sleeping
Sleeping
MilesCranmer
commited on
Commit
•
9ff66c9
1
Parent(s):
b8953fa
Change default niterations to 40
Browse files- README.md +1 -1
- example.py +1 -1
- pysr/sr.py +1 -1
README.md
CHANGED
@@ -87,7 +87,7 @@ PySR's main interface is in the style of scikit-learn:
|
|
87 |
```python
|
88 |
from pysr import PySRRegressor
|
89 |
model = PySRRegressor(
|
90 |
-
niterations=
|
91 |
binary_operators=["+", "*"],
|
92 |
unary_operators=[
|
93 |
"cos",
|
|
|
87 |
```python
|
88 |
from pysr import PySRRegressor
|
89 |
model = PySRRegressor(
|
90 |
+
niterations=40,
|
91 |
binary_operators=["+", "*"],
|
92 |
unary_operators=[
|
93 |
"cos",
|
example.py
CHANGED
@@ -6,7 +6,7 @@ y = 2.5382 * np.cos(X[:, 3]) + X[:, 0] ** 2 - 0.5
|
|
6 |
from pysr import PySRRegressor
|
7 |
|
8 |
model = PySRRegressor(
|
9 |
-
niterations=
|
10 |
binary_operators=["+", "*"],
|
11 |
unary_operators=[
|
12 |
"cos",
|
|
|
6 |
from pysr import PySRRegressor
|
7 |
|
8 |
model = PySRRegressor(
|
9 |
+
niterations=40,
|
10 |
binary_operators=["+", "*"],
|
11 |
unary_operators=[
|
12 |
"cos",
|
pysr/sr.py
CHANGED
@@ -351,7 +351,7 @@ class PySRRegressor(BaseEstimator, RegressorMixin):
|
|
351 |
procs=cpu_count(),
|
352 |
loss="L2DistLoss()",
|
353 |
populations=15,
|
354 |
-
niterations=
|
355 |
ncyclesperiteration=550,
|
356 |
timeout_in_seconds=None,
|
357 |
alpha=0.1,
|
|
|
351 |
procs=cpu_count(),
|
352 |
loss="L2DistLoss()",
|
353 |
populations=15,
|
354 |
+
niterations=40,
|
355 |
ncyclesperiteration=550,
|
356 |
timeout_in_seconds=None,
|
357 |
alpha=0.1,
|