Latent Dirichlet Allocation

Imperative implementation of a collapsed Gibbs sampler for LDA. This library uses the topic modeling terminology (documents, words, topics), even though it is generic. For example if used for word class induction, replace documents with word types, words with features and topics with word classes.

Usage example:

 import qualified Data.Vector as V
 import qualified Data.Vector.Unboxed as U
 import  Control.Monad.ST
 import NLP.SwiftLDA
 
 main = do
   -- Initialize model.
   let river = 1
       money = 2
       bank  = 3
   m <- stToIO $ initial (U.singleton 42) 2 10 0.1
   let  docs = [ (1, U.fromList [river, river, bank])  
               , (2, U.fromList [money, money, bank])
               , (3, U.fromList [river, money, bank])
               ]
   -- Run 10 iterations of sampling on this batch of documents.
   result <- stToIO $ run m 10 docs
   -- Display topic assignments.
   print result
   -- Run one iteration of sampling on another batch and display result
   print =<< (stToIO $ run m 1 [(4, U.fromList [bank, bank])]) 
   -- Retrieve and display word-topic weights.
   fm <- stToIO $ finalize m
   print $ wordTopics fm