MilesCranmer commited on
Commit
5abd46e
1 Parent(s): f422d37

Bump backend version with `should_simplify` param

Browse files
Files changed (3) hide show
  1. docs/param_groupings.yml +1 -0
  2. pysr/sr.py +6 -0
  3. pysr/version.py +2 -2
docs/param_groupings.yml CHANGED
@@ -23,6 +23,7 @@
23
  - use_frequency
24
  - use_frequency_in_tournament
25
  - adaptive_parsimony_scaling
 
26
  - Mutations:
27
  - weight_add_node
28
  - weight_insert_node
 
23
  - use_frequency
24
  - use_frequency_in_tournament
25
  - adaptive_parsimony_scaling
26
+ - should_simplify
27
  - Mutations:
28
  - weight_add_node
29
  - weight_insert_node
pysr/sr.py CHANGED
@@ -436,6 +436,9 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
436
  topn : int
437
  How many top individuals migrate from each population.
438
  Default is `12`.
 
 
 
439
  should_optimize_constants : bool
440
  Whether to numerically optimize constants (Nelder-Mead/Newton)
441
  at the end of each iteration. Default is `True`.
@@ -706,6 +709,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
706
  migration=True,
707
  hof_migration=True,
708
  topn=12,
 
709
  should_optimize_constants=True,
710
  optimizer_algorithm="BFGS",
711
  optimizer_nrestarts=2,
@@ -759,6 +763,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
759
  self.constraints = constraints
760
  self.nested_constraints = nested_constraints
761
  self.warmup_maxsize_by = warmup_maxsize_by
 
762
  # - Early exit conditions:
763
  self.max_evals = max_evals
764
  self.timeout_in_seconds = timeout_in_seconds
@@ -1593,6 +1598,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1593
  migration=self.migration,
1594
  hof_migration=self.hof_migration,
1595
  fraction_replaced_hof=self.fraction_replaced_hof,
 
1596
  should_optimize_constants=self.should_optimize_constants,
1597
  warmup_maxsize_by=self.warmup_maxsize_by,
1598
  use_frequency=self.use_frequency,
 
436
  topn : int
437
  How many top individuals migrate from each population.
438
  Default is `12`.
439
+ should_simplify : bool
440
+ Whether to use algebraic simplification in the search. Note that only
441
+ a few simple rules are implemented. Default is `True`.
442
  should_optimize_constants : bool
443
  Whether to numerically optimize constants (Nelder-Mead/Newton)
444
  at the end of each iteration. Default is `True`.
 
709
  migration=True,
710
  hof_migration=True,
711
  topn=12,
712
+ should_simplify=None,
713
  should_optimize_constants=True,
714
  optimizer_algorithm="BFGS",
715
  optimizer_nrestarts=2,
 
763
  self.constraints = constraints
764
  self.nested_constraints = nested_constraints
765
  self.warmup_maxsize_by = warmup_maxsize_by
766
+ self.should_simplify = should_simplify
767
  # - Early exit conditions:
768
  self.max_evals = max_evals
769
  self.timeout_in_seconds = timeout_in_seconds
 
1598
  migration=self.migration,
1599
  hof_migration=self.hof_migration,
1600
  fraction_replaced_hof=self.fraction_replaced_hof,
1601
+ should_simplify=self.should_simplify,
1602
  should_optimize_constants=self.should_optimize_constants,
1603
  warmup_maxsize_by=self.warmup_maxsize_by,
1604
  use_frequency=self.use_frequency,
pysr/version.py CHANGED
@@ -1,2 +1,2 @@
1
- __version__ = "0.12.0"
2
- __symbolic_regression_jl_version__ = "0.16.1"
 
1
+ __version__ = "0.12.1"
2
+ __symbolic_regression_jl_version__ = "0.16.2"