Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
fa9d281
1
Parent(s):
75c23d4
docs: update __getstate__ docstring
Browse files- pysr/sr.py +2 -9
pysr/sr.py
CHANGED
@@ -1057,15 +1057,8 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
|
|
1057 |
Handle pickle serialization for PySRRegressor.
|
1058 |
|
1059 |
The Scikit-learn standard requires estimators to be serializable via
|
1060 |
-
`pickle.dumps()`. However,
|
1061 |
-
|
1062 |
-
|
1063 |
-
Thus, for `PySRRegressor` to support pickle serialization, the
|
1064 |
-
`julia_state_stream_` attribute must be hidden from pickle. This will
|
1065 |
-
prevent the `warm_start` of any model that is loaded via `pickle.loads()`,
|
1066 |
-
but does allow all other attributes of a fitted `PySRRegressor` estimator
|
1067 |
-
to be serialized. Note: Jax and Torch format equations are also removed
|
1068 |
-
from the pickled instance.
|
1069 |
"""
|
1070 |
state = self.__dict__
|
1071 |
show_pickle_warning = not (
|
|
|
1057 |
Handle pickle serialization for PySRRegressor.
|
1058 |
|
1059 |
The Scikit-learn standard requires estimators to be serializable via
|
1060 |
+
`pickle.dumps()`. However, some attributes do not support pickling
|
1061 |
+
and need to be hidden, such as the JAX and Torch representations.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1062 |
"""
|
1063 |
state = self.__dict__
|
1064 |
show_pickle_warning = not (
|