Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
c526fa1
1
Parent(s):
551b15e
Add turbo mode
Browse files- pysr/sr.py +6 -0
- 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.
|
|
|
1 |
__version__ = "0.11.6"
|
2 |
+
__symbolic_regression_jl_version__ = "0.14.0"
|