From 89e991d9d7a5e34025bdbbe9076a8fd0892f04cc Mon Sep 17 00:00:00 2001 From: MilesCranmer Date: Sun, 16 Jun 2024 18:02:37 +0100 Subject: [PATCH] docs: improve `fit` docstring --- pysr/sr.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pysr/sr.py b/pysr/sr.py index 8ccfb79cc..916ae8cd8 100644 --- a/pysr/sr.py +++ b/pysr/sr.py @@ -1394,13 +1394,13 @@ def _validate_and_set_fit_params( Each element is how to weight the mean-square-error loss for that particular element of y. variable_names : ndarray of length n_features - Names of each variable in the training dataset, `X`. + Names of each feature in the training dataset, `X`. complexity_of_variables : int | float | list[int | float] - Complexity of each variable in the training dataset, `X`. + Complexity of each feature in the training dataset, `X`. X_units : list[str] of length n_features - Units of each variable in the training dataset, `X`. + Units of each feature in the training dataset, `X`. y_units : str | list[str] of length n_out - Units of each variable in the training dataset, `y`. + Units of each feature in the training dataset, `y`. Returns -------