@hackage / welford-online-mean-variance

Online computation of mean and variance using the Welford algorithm.

Latest0.2.0.0

About

Metadata

  • Last updated , by schnecki
  • License BSD-3-Clause

  • Maintained by: manuel.schnecki@gmail.com

  • Lottery factor: 0

Links

Installation

Readme

Welford: Online mean and variance computation

Example

example :: [Double] -> IO ()
example vals = do
  let n = fromIntegral (length vals)
      mean = sum vals / n
      var = sum (map (\x -> (x - mean) ^ 2) vals) / (n - 1)
      (wMean, _, wVarSample) = finalize $ foldl' addValue WelfordExistingAggregateEmpty vals
  print (mean, var)
  print (wMean, wVarSample)

WelfordExistingAggregate is used to save the state. Use the function finalize to retrieve the current estimates for the mean, variance and sample variance.