MilesCranmer commited on
Commit
a47d265
1 Parent(s): 38e582f

Make default model selection = 'best'

Browse files
Files changed (1) hide show
  1. 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="accuracy",
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