Spaces:

MilesCranmer
/

PySR

Sleeping

MilesCranmer commited on Jan 16, 2022

Commit

79ef9ab

1 Parent(s): 501ebd3

Fix bug with curMSE equal to 0.0

Files changed (1) hide show

pysr/sr.py CHANGED Viewed

@@ -1067,7 +1067,10 @@ def get_hof(
             if lastMSE is None:
                 cur_score = 0.0
             else:
-                cur_score = -np.log(curMSE / lastMSE) / (curComplexity - lastComplexity)
             scores.append(cur_score)
             lastMSE = curMSE

             if lastMSE is None:
                 cur_score = 0.0
             else:
+                if curMSE > 0.0:
+                    cur_score = -np.log(curMSE / lastMSE) / (curComplexity - lastComplexity)
+                else:
+                    cur_score = np.inf
             scores.append(cur_score)
             lastMSE = curMSE