Spaces:
Running
Running
Fix `julia_state_` typo
Browse files- pysr/param_groupings.yml +0 -1
- pysr/sr.py +4 -2
pysr/param_groupings.yml
CHANGED
|
@@ -89,7 +89,6 @@
|
|
| 89 |
- tempdir
|
| 90 |
- delete_tempfiles
|
| 91 |
- update
|
| 92 |
-
- julia_kwargs
|
| 93 |
- Exporting the Results:
|
| 94 |
- equation_file
|
| 95 |
- output_jax_format
|
|
|
|
| 89 |
- tempdir
|
| 90 |
- delete_tempfiles
|
| 91 |
- update
|
|
|
|
| 92 |
- Exporting the Results:
|
| 93 |
- equation_file
|
| 94 |
- output_jax_format
|
pysr/sr.py
CHANGED
|
@@ -605,6 +605,8 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
|
|
| 605 |
raw_julia_state_stream_ : ndarray
|
| 606 |
The serialized state for the julia SymbolicRegression.jl backend (after fitting),
|
| 607 |
stored as an array of uint8, produced by Julia's Serialization.serialize function.
|
|
|
|
|
|
|
| 608 |
equation_file_contents_ : list[pandas.DataFrame]
|
| 609 |
Contents of the equation file output by the Julia backend.
|
| 610 |
show_pickle_warnings_ : bool
|
|
@@ -1131,7 +1133,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
|
|
| 1131 |
"for the raw stream of bytes.",
|
| 1132 |
FutureWarning,
|
| 1133 |
)
|
| 1134 |
-
return self.
|
| 1135 |
|
| 1136 |
def get_best(self, index=None):
|
| 1137 |
"""
|
|
@@ -1735,7 +1737,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
|
|
| 1735 |
options=options,
|
| 1736 |
numprocs=cprocs,
|
| 1737 |
parallelism=parallelism,
|
| 1738 |
-
saved_state=self.
|
| 1739 |
return_state=True,
|
| 1740 |
addprocs_function=cluster_manager,
|
| 1741 |
heap_size_hint_in_bytes=self.heap_size_hint_in_bytes,
|
|
|
|
| 605 |
raw_julia_state_stream_ : ndarray
|
| 606 |
The serialized state for the julia SymbolicRegression.jl backend (after fitting),
|
| 607 |
stored as an array of uint8, produced by Julia's Serialization.serialize function.
|
| 608 |
+
julia_state_ : ndarray
|
| 609 |
+
The deserialized state.
|
| 610 |
equation_file_contents_ : list[pandas.DataFrame]
|
| 611 |
Contents of the equation file output by the Julia backend.
|
| 612 |
show_pickle_warnings_ : bool
|
|
|
|
| 1133 |
"for the raw stream of bytes.",
|
| 1134 |
FutureWarning,
|
| 1135 |
)
|
| 1136 |
+
return self.julia_state_
|
| 1137 |
|
| 1138 |
def get_best(self, index=None):
|
| 1139 |
"""
|
|
|
|
| 1737 |
options=options,
|
| 1738 |
numprocs=cprocs,
|
| 1739 |
parallelism=parallelism,
|
| 1740 |
+
saved_state=self.julia_state_,
|
| 1741 |
return_state=True,
|
| 1742 |
addprocs_function=cluster_manager,
|
| 1743 |
heap_size_hint_in_bytes=self.heap_size_hint_in_bytes,
|