Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
a47d265
1
Parent(s):
38e582f
Make default model selection = 'best'
Browse files- pysr/sr.py +2 -2
pysr/sr.py
CHANGED
@@ -317,7 +317,7 @@ julia = "1.5"
|
|
317 |
class PySRRegressor(BaseEstimator, RegressorMixin):
|
318 |
def __init__(
|
319 |
self,
|
320 |
-
model_selection="
|
321 |
weights=None,
|
322 |
binary_operators=None,
|
323 |
unary_operators=None,
|
@@ -388,7 +388,7 @@ class PySRRegressor(BaseEstimator, RegressorMixin):
|
|
388 |
You can view more detailed explanations of the options on the
|
389 |
[options page](https://pysr.readthedocs.io/en/latest/docs/options/) of the documentation.
|
390 |
|
391 |
-
:param model_selection: How to select a model. Can be 'accuracy' or 'best'. 'best' will optimize a combination of complexity and accuracy.
|
392 |
:type model_selection: str
|
393 |
:param binary_operators: List of strings giving the binary operators in Julia's Base. Default is ["+", "-", "*", "/",].
|
394 |
:type binary_operators: list
|
|
|
317 |
class PySRRegressor(BaseEstimator, RegressorMixin):
|
318 |
def __init__(
|
319 |
self,
|
320 |
+
model_selection="best",
|
321 |
weights=None,
|
322 |
binary_operators=None,
|
323 |
unary_operators=None,
|
|
|
388 |
You can view more detailed explanations of the options on the
|
389 |
[options page](https://pysr.readthedocs.io/en/latest/docs/options/) of the documentation.
|
390 |
|
391 |
+
:param model_selection: How to select a model. Can be 'accuracy' or 'best'. The default, 'best', will optimize a combination of complexity and accuracy.
|
392 |
:type model_selection: str
|
393 |
:param binary_operators: List of strings giving the binary operators in Julia's Base. Default is ["+", "-", "*", "/",].
|
394 |
:type binary_operators: list
|