Spaces:
Running
Running
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"
|