Safe Haskell | None |
---|
Documentation
A conditional random field model with additional codec used for data encoding.
:: (Ord a, Ord b) | |
=> SgdArgs | Args for SGD |
-> IO [SentL a b] | Training data |
-> Maybe (IO [SentL a b]) | Maybe evalation data |
-> (AVec Lb -> [(Xs, Ys)] -> [Feature]) | Feature selection |
-> IO (CRF a b) | Resulting model |
Train the CRF using the stochastic gradient descent method.
The resulting model will contain features extracted with
the user supplied extraction function.
You can use the functions provided by the Data.CRF.Chain1.Feature.Present
and Data.CRF.Chain1.Feature.Hidden modules for this purpose.
When the evaluation data IO
action is Just
, the iterative
training process will notify the user about the current accuracy
on the evaluation part every full iteration over the training part.
TODO: Accept custom r0 construction function.