MilesCranmer commited on
Commit
91a0f71
1 Parent(s): 895da91

Improve printed string

Browse files
Files changed (1) hide show
  1. pysr/sklearn.py +3 -3
pysr/sklearn.py CHANGED
@@ -30,11 +30,11 @@ class PySRRegressor(BaseEstimator, RegressorMixin):
30
  chosen_row = equations["score"].idxmax()
31
  else:
32
  raise NotImplementedError
33
- selected[chosen_row] = ">"
34
- output = "PySRRegressor.equations=[\n"
35
  repr_equations = pd.DataFrame(
36
  dict(
37
- selected=selected,
38
  score=equations["score"],
39
  Equation=equations["Equation"],
40
  MSE=equations["MSE"],
 
30
  chosen_row = equations["score"].idxmax()
31
  else:
32
  raise NotImplementedError
33
+ selected[chosen_row] = ">>>>"
34
+ output = "PySRRegressor.equations = [\n"
35
  repr_equations = pd.DataFrame(
36
  dict(
37
+ pick=selected,
38
  score=equations["score"],
39
  Equation=equations["Equation"],
40
  MSE=equations["MSE"],