MilesCranmer commited on
Commit
6db5312
1 Parent(s): d0ea029

Remove mention of `select_k_features` in feature warning

Browse files
Files changed (1) hide show
  1. pysr/sr.py +3 -3
pysr/sr.py CHANGED
@@ -1761,9 +1761,9 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1761
  warnings.warn(
1762
  "Note: you are running with 10 features or more. "
1763
  "Genetic algorithms like used in PySR scale poorly with large numbers of features. "
1764
- "Consider using feature selection techniques to select the most important features "
1765
- "(you can do this automatically with the `select_k_features` parameter), "
1766
- "or, alternatively, doing a dimensionality reduction beforehand. "
1767
  "For example, `X = PCA(n_components=6).fit_transform(X)`, "
1768
  "using scikit-learn's `PCA` class, "
1769
  "will reduce the number of features to 6 in an interpretable way, "
 
1761
  warnings.warn(
1762
  "Note: you are running with 10 features or more. "
1763
  "Genetic algorithms like used in PySR scale poorly with large numbers of features. "
1764
+ "You should run PySR for more `niterations` to ensure it can find "
1765
+ "the correct variables, "
1766
+ "or, alternatively, do a dimensionality reduction beforehand. "
1767
  "For example, `X = PCA(n_components=6).fit_transform(X)`, "
1768
  "using scikit-learn's `PCA` class, "
1769
  "will reduce the number of features to 6 in an interpretable way, "