Fix DomainError in pareto curve; fixes #67

MilesCranmer · Feb 2, 2022 · ded3037 · ded3037 · MilesCranmer · Feb 2, 2022
1 parent 215900f
commit ded3037
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "SymbolicRegression"
 uuid = "8254be44-1295-4e6a-a16d-46603ac705cb"
 authors = ["MilesCranmer <miles.cranmer@gmail.com>"]
-version = "0.7.5"
+version = "0.7.6"
 
 [deps]
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"

diff --git a/src/HallOfFame.jl b/src/HallOfFame.jl
@@ -97,7 +97,8 @@ function string_dominating_pareto_curve(hallOfFame, baselineMSE,
         complexity = countNodes(member.tree)
         curMSE = (member.score - complexity * options.parsimony) * baselineMSE
         delta_c = complexity - lastComplexity
-        delta_l_mse = log(curMSE/lastMSE)
+        ZERO_POINT = 1f-10
+        delta_l_mse = log(abs(curMSE/lastMSE) + ZERO_POINT)
         score = convert(Float32, -delta_l_mse/delta_c)
         output *= @sprintf("%-10d  %-8.3e  %-8.3e  %-s\n" , complexity, curMSE, score, stringTree(member.tree, options, varMap=dataset.varMap))
         lastMSE = curMSE