MilesCranmer commited on
Commit
89e991d
1 Parent(s): c2ab38b

docs: improve `fit` docstring

Browse files
Files changed (1) hide show
  1. pysr/sr.py +4 -4
pysr/sr.py CHANGED
@@ -1394,13 +1394,13 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1394
  Each element is how to weight the mean-square-error loss
1395
  for that particular element of y.
1396
  variable_names : ndarray of length n_features
1397
- Names of each variable in the training dataset, `X`.
1398
  complexity_of_variables : int | float | list[int | float]
1399
- Complexity of each variable in the training dataset, `X`.
1400
  X_units : list[str] of length n_features
1401
- Units of each variable in the training dataset, `X`.
1402
  y_units : str | list[str] of length n_out
1403
- Units of each variable in the training dataset, `y`.
1404
 
1405
  Returns
1406
  -------
 
1394
  Each element is how to weight the mean-square-error loss
1395
  for that particular element of y.
1396
  variable_names : ndarray of length n_features
1397
+ Names of each feature in the training dataset, `X`.
1398
  complexity_of_variables : int | float | list[int | float]
1399
+ Complexity of each feature in the training dataset, `X`.
1400
  X_units : list[str] of length n_features
1401
+ Units of each feature in the training dataset, `X`.
1402
  y_units : str | list[str] of length n_out
1403
+ Units of each feature in the training dataset, `y`.
1404
 
1405
  Returns
1406
  -------