Safe Haskell | None |
---|---|
Language | Haskell98 |
A version of SGD
extended with momentum.
Synopsis
Documentation
SGD parameters controlling the learning process.
sgdArgsDefault :: SgdArgs Source #
Default SGD parameter values.
:: SgdArgs | SGD parameter values |
-> (Para -> Int -> IO ()) | Notification run every update |
-> (Para -> x -> Grad) | Gradient for dataset element |
-> Dataset x | Dataset |
-> Para | Starting point |
-> IO Para | SGD result |
A stochastic gradient descent method. A notification function can be used to provide user with information about the progress of the learning.
module Numeric.SGD.Grad
module Numeric.SGD.Dataset