quantified.uncertainty/squiggle
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: change integrand check order in KL divergence code

Browse Source
This commit is contained in:
NunoSempere 2022-05-06 09:49:42 -04:00
parent fd17da83b8
commit 87d6f8d4f6
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
packages/squiggle-lang/src/rescript/Distributions/PointSetDist/PointSetDist_Scoring.res
View File

@ -4,10 +4,10 @@ module KLDivergence = {
float, float,
Operation.Error.t, Operation.Error.t,
> => > =>
if predictionElement == 0.0 { if answerElement == 0.0 {
Error(Operation.NegativeInfinityError)
} else if answerElement == 0.0 {
Ok(0.0) Ok(0.0)
} else if predictionElement == 0.0 {
Error(Operation.NegativeInfinityError)
} else { } else {
let quot = predictionElement /. answerElement let quot = predictionElement /. answerElement
quot < 0.0 ? Error(Operation.ComplexNumberError) : Ok(answerElement *. logFn(quot)) quot < 0.0 ? Error(Operation.ComplexNumberError) : Ok(answerElement *. logFn(quot))