Merge pull request #36 from cobac/master

Allow infinities in recorder
MilesCranmer · Jun 22, 2021 · 65dbd0e · 65dbd0e
2 parents 124020d + ddf252e
commit 65dbd0e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/Population.jl b/src/Population.jl
@@ -89,9 +89,8 @@ end
 
 
 function record_population(pop::Population{T}, options::Options)::RecordType where {T<:Real}
-    inf_to_num = x -> isinf(x) ? convert(T, 1e9) : x
     RecordType("population"=>[RecordType("tree"=>stringTree(member.tree, options),
-                                         "loss"=>inf_to_num(member.score),
+                                         "loss"=>member.score,
                                          "complexity"=>countNodes(member.tree),
                                          "birth"=>member.birth,
                                          "ref"=>member.ref,

diff --git a/src/SymbolicRegression.jl b/src/SymbolicRegression.jl
@@ -569,7 +569,7 @@ function _EquationSearch(::ConcurrencyType, datasets::Array{Dataset{T}, 1};
 
     @recorder begin
         open(options.recorder_file, "w") do io
-            JSON3.write(io, record)
+            JSON3.write(io, record, allow_inf = true)
         end
     end