Skip to content

Commit

Permalink
Remove mention of select_k_features in feature warning
Browse files Browse the repository at this point in the history
  • Loading branch information
MilesCranmer committed Mar 13, 2023
1 parent d0ea029 commit 6db5312
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions pysr/sr.py
Original file line number Diff line number Diff line change
Expand Up @@ -1761,9 +1761,9 @@ def fit(
warnings.warn(
"Note: you are running with 10 features or more. "
"Genetic algorithms like used in PySR scale poorly with large numbers of features. "
"Consider using feature selection techniques to select the most important features "
"(you can do this automatically with the `select_k_features` parameter), "
"or, alternatively, doing a dimensionality reduction beforehand. "
"You should run PySR for more `niterations` to ensure it can find "
"the correct variables, "
"or, alternatively, do a dimensionality reduction beforehand. "
"For example, `X = PCA(n_components=6).fit_transform(X)`, "
"using scikit-learn's `PCA` class, "
"will reduce the number of features to 6 in an interpretable way, "
Expand Down

0 comments on commit 6db5312

Please sign in to comment.