MilesCranmer commited on
Commit
9ff66c9
1 Parent(s): b8953fa

Change default niterations to 40

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. example.py +1 -1
  3. 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=5,
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=5,
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=4,
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,