BayesStack.DirMulti

Dirichlet/multinomial pair

data Multinom a

'Multinom a' represents multinomial distribution over domain a. Optionally, this can include a collapsed Dirichlet prior. 'Multinom alpha count total' is a multinomial with Dirichlet prior with symmetric parameter alpha, ...

Instances

FullConditionable Multinom

HasLikelihood Multinom

Eq a => Eq (Multinom a)

(Enum a, Show a) => Show (Multinom a)

Generic (Multinom a)

(Enum a, Serialize a) => Serialize (Multinom a)

dirMulti :: Enum a => [(a, Double)] -> Multinom a

Create an asymmetric Dirichlet/multinomial from items and alphas

symDirMulti :: Enum a => Double -> [a] -> Multinom a

Create a symmetric Dirichlet/multinomial

multinom :: Enum a => [(a, Double)] -> Multinom a

A multinomial without a prior

Do not do record updates with these

dmTotal :: Multinom a -> Int

dmAlpha :: Multinom a -> Alpha a

dmDomain :: Multinom a -> Seq a

setMultinom :: (Enum a, Ord a) => SetUnset -> a -> Multinom a -> Multinom a

data SetUnset

Constructors

Set

Unset

decMultinom :: (Ord a, Enum a) => a -> Multinom a -> Multinom a

incMultinom :: (Ord a, Enum a) => a -> Multinom a -> Multinom a

prettyMultinom :: (Ord a, Enum a) => Int -> (a -> String) -> Multinom a -> Doc

updatePrior :: (Alpha a -> Alpha a) -> Multinom a -> Multinom a

Update the prior of a Dirichlet/multinomial

Parameter estimation

estimatePrior :: Enum a => Double -> [Multinom a] -> Alpha a

reestimatePriors :: (Foldable f, Functor f, Enum a) => f (Multinom a) -> f (Multinom a)

reestimateSymPriors :: (Foldable f, Functor f, Enum a) => f (Multinom a) -> f (Multinom a)

Convenience functions

probabilities :: (Ord a, Enum a) => Multinom a -> Seq (Double, a)

decProbabilities :: (Ord a, Enum a) => Multinom a -> Seq (Double, a)

Probabilities sorted decreasingly