System.Random.MWC.Probability

A probability monad based on sampling functions.

Probability distributions are abstract constructs that can be represented in a variety of ways. The sampling function representation is particularly useful - it's computationally efficient, and collections of samples are amenable to much practical work.

Probability monads propagate uncertainty under the hood. An expression like beta 1 8 >>= binomial 10 corresponds to a beta-binomial distribution in which the uncertainty captured by beta 1 8 has been marginalized out.

The distribution resulting from a series of effects is called the predictive distribution of the model described by the corresponding expression. The monadic structure lets one piece together a hierarchical structure from simpler, local conditionals:

hierarchicalModel = do
  [c, d, e, f] <- replicateM 4 $ uniformR (1, 10)
  a <- gamma c d
  b <- gamma e f
  p <- beta a b
  n <- uniformR (5, 10)
  binomial n p

The functor instance for a probability monad transforms the support of the distribution while leaving its density structure invariant in some sense. For example, uniform is a distribution over the 0-1 interval, but fmap (+ 1) uniform is the translated distribution over the 1-2 interval.

>>> sample (fmap (+ 1) uniform) gen
1.5480073474340754

Documentation

newtype Prob m a

A probability distribution characterized by a sampling function.

>>> gen <- create
>>> sample uniform gen
0.4208881170464097

Constructors

Prob

Fields

• sample :: Gen (PrimState m) -> m a

Instances

MonadTrans Prob

Methods

lift :: Monad m => m a -> Prob m a

Monad m => Monad (Prob m)

Methods

(>>=) :: Prob m a -> (a -> Prob m b) -> Prob m b

(>>) :: Prob m a -> Prob m b -> Prob m b

return :: a -> Prob m a

fail :: String -> Prob m a

Monad m => Functor (Prob m)

Methods

fmap :: (a -> b) -> Prob m a -> Prob m b

(<$) :: a -> Prob m b -> Prob m a

Monad m => Applicative (Prob m)

Methods

pure :: a -> Prob m a

(<*>) :: Prob m (a -> b) -> Prob m a -> Prob m b

(*>) :: Prob m a -> Prob m b -> Prob m b

(<*) :: Prob m a -> Prob m b -> Prob m a

MonadIO m => MonadIO (Prob m)

Methods

liftIO :: IO a -> Prob m a

PrimMonad m => PrimMonad (Prob m)

Associated Types

type PrimState (Prob m :: * -> *) :: *

Methods

primitive :: (State# (PrimState (Prob m)) -> (#VoidRep, PtrRepLifted, State# (PrimState (Prob m)), a#)) -> Prob m a

(Monad m, Num a) => Num (Prob m a)

Methods

(+) :: Prob m a -> Prob m a -> Prob m a

(-) :: Prob m a -> Prob m a -> Prob m a

(*) :: Prob m a -> Prob m a -> Prob m a

negate :: Prob m a -> Prob m a

abs :: Prob m a -> Prob m a

signum :: Prob m a -> Prob m a

fromInteger :: Integer -> Prob m a

type PrimState (Prob m)

type PrimState (Prob m) = PrimState m

samples :: PrimMonad m => Int -> Prob m a -> Gen (PrimState m) -> m [a]

Sample from a model n times.

>>> samples 2 uniform gen
[0.6738707766845254,0.9730405951541817]

uniform :: (PrimMonad m, Variate a) => Prob m a

The uniform distribution over a type.

>>> gen <- create
>>> sample uniform gen :: IO Double
0.29308497534914946
>>> sample uniform gen :: IO Bool
False

uniformR :: (PrimMonad m, Variate a) => (a, a) -> Prob m a

The uniform distribution over the provided interval.

>>> sample (uniformR (0, 1)) gen
0.44984153252922365

discreteUniform :: (PrimMonad m, Foldable f) => f a -> Prob m a

The discrete uniform distribution.

>>> sample (discreteUniform [0..10]) gen
6
>>> sample (discreteUniform "abcdefghijklmnopqrstuvwxyz") gen
'a'

categorical :: (Foldable f, PrimMonad m) => f Double -> Prob m Int

A categorical distribution defined by the supplied list of probabilities.

standard :: PrimMonad m => Prob m Double

The standard normal or Gaussian distribution (with mean 0 and standard deviation 1).

normal :: PrimMonad m => Double -> Double -> Prob m Double

The normal or Gaussian distribution with a specified mean and standard deviation.

logNormal :: PrimMonad m => Double -> Double -> Prob m Double

The log-normal distribution with specified mean and standard deviation.

exponential :: PrimMonad m => Double -> Prob m Double

The exponential distribution with provided rate parameter.

gamma :: PrimMonad m => Double -> Double -> Prob m Double

The gamma distribution with shape parameter a and scale parameter b.

This is the parameterization used more traditionally in frequentist statistics. It has the following corresponding probability density function:

f(x; a, b) = 1 (Gamma(a) * b ^ a) x ^ (a - 1) e ^ (- x b)

inverseGamma :: PrimMonad m => Double -> Double -> Prob m Double

The inverse-gamma distribution.

chiSquare :: PrimMonad m => Int -> Prob m Double

The chi-square distribution.

beta :: PrimMonad m => Double -> Double -> Prob m Double

The beta distribution.

dirichlet :: (Traversable f, PrimMonad m) => f Double -> Prob m (f Double)

The Dirichlet distribution.

symmetricDirichlet :: PrimMonad m => Int -> Double -> Prob m [Double]

The symmetric Dirichlet distribution of dimension n.

bernoulli :: PrimMonad m => Double -> Prob m Bool

The Bernoulli distribution.

binomial :: PrimMonad m => Int -> Double -> Prob m Int

The binomial distribution.

multinomial :: (Foldable f, PrimMonad m) => Int -> f Double -> Prob m [Int]

The multinomial distribution.

student :: PrimMonad m => Double -> Double -> Double -> Prob m Double

Student's t distribution.

isoGauss :: (Traversable f, PrimMonad m) => f Double -> Double -> Prob m (f Double)

An isotropic or spherical Gaussian distribution.

poisson :: PrimMonad m => Double -> Prob m Int

The Poisson distribution.