diff options
| -rw-r--r-- | src/PPL/Internal.hs | 4 | ||||
| -rw-r--r-- | src/PPL/Sampling.hs | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/PPL/Internal.hs b/src/PPL/Internal.hs index 201a3aa..d927508 100644 --- a/src/PPL/Internal.hs +++ b/src/PPL/Internal.hs @@ -18,7 +18,7 @@ import Control.Monad.IO.Class -- Reimplementation of the LazyPPL monads to avoid some dependencies -data Tree = Tree Double [Tree] +data Tree = Tree !Double [Tree] split :: Tree -> (Tree, Tree) split (Tree r (t : ts)) = (t, Tree r ts) @@ -61,7 +61,7 @@ newtype Meas a = Meas (WriterT (Product (Log Double)) Prob a) score :: Double -> Meas () score = scoreLog . Exp . log . max eps where - eps = $(TH.lift (until ((== 1) . (1 +)) (/ 2) (1 :: Double))) -- machine epsilon, force compile time eval + eps = $(TH.lift (2 * until ((== 1) . (1 +)) (/ 2) (1 :: Double))) -- machine epsilon, force compile time eval {-# INLINE scoreLog #-} scoreLog :: Log Double -> Meas () diff --git a/src/PPL/Sampling.hs b/src/PPL/Sampling.hs index a3f38db..1d20838 100644 --- a/src/PPL/Sampling.hs +++ b/src/PPL/Sampling.hs @@ -42,7 +42,7 @@ mh p m = do ratio = w' / w (Exp . log -> r, g3) = random g2 put g3 - pure $ + pure $! if r < ratio then (t', x', w') else (t, x, w) |