MilesCranmer commited on
Commit
c526fa1
1 Parent(s): 551b15e

Add turbo mode

Browse files
Files changed (2) hide show
  1. pysr/sr.py +6 -0
  2. pysr/version.py +1 -1
pysr/sr.py CHANGED
@@ -476,6 +476,9 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
476
  algorithm than regularized evolution, but does cycles 15%
477
  faster. May be algorithmically less efficient.
478
  Default is `False`.
 
 
 
479
  precision : int
480
  What precision to use for the data. By default this is `32`
481
  (float32), but you can select `64` or `16` as well, giving
@@ -692,6 +695,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
692
  batching=False,
693
  batch_size=50,
694
  fast_cycle=False,
 
695
  precision=32,
696
  random_state=None,
697
  deterministic=False,
@@ -779,6 +783,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
779
  self.batching = batching
780
  self.batch_size = batch_size
781
  self.fast_cycle = fast_cycle
 
782
  self.precision = precision
783
  self.random_state = random_state
784
  self.deterministic = deterministic
@@ -1554,6 +1559,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1554
  alpha=self.alpha,
1555
  maxdepth=maxdepth,
1556
  fast_cycle=self.fast_cycle,
 
1557
  migration=self.migration,
1558
  hof_migration=self.hof_migration,
1559
  fraction_replaced_hof=self.fraction_replaced_hof,
 
476
  algorithm than regularized evolution, but does cycles 15%
477
  faster. May be algorithmically less efficient.
478
  Default is `False`.
479
+ turbo: bool
480
+ Whether to use LoopVectorization.jl to speed up the search
481
+ evaluation. Certain user-defined operators may not be supported.
482
  precision : int
483
  What precision to use for the data. By default this is `32`
484
  (float32), but you can select `64` or `16` as well, giving
 
695
  batching=False,
696
  batch_size=50,
697
  fast_cycle=False,
698
+ turbo=False,
699
  precision=32,
700
  random_state=None,
701
  deterministic=False,
 
783
  self.batching = batching
784
  self.batch_size = batch_size
785
  self.fast_cycle = fast_cycle
786
+ self.turbo = turbo
787
  self.precision = precision
788
  self.random_state = random_state
789
  self.deterministic = deterministic
 
1559
  alpha=self.alpha,
1560
  maxdepth=maxdepth,
1561
  fast_cycle=self.fast_cycle,
1562
+ turbo=self.turbo,
1563
  migration=self.migration,
1564
  hof_migration=self.hof_migration,
1565
  fraction_replaced_hof=self.fraction_replaced_hof,
pysr/version.py CHANGED
@@ -1,2 +1,2 @@
1
  __version__ = "0.11.6"
2
- __symbolic_regression_jl_version__ = "0.13.2"
 
1
  __version__ = "0.11.6"
2
+ __symbolic_regression_jl_version__ = "0.14.0"