RosettaCodeData/Task/Polynomial-regression/Haskell/polynomial-regression-1.hs
2023-07-01 13:44:08 -04:00

11 lines
290 B
Haskell

import Data.List
import Data.Array
import Control.Monad
import Control.Arrow
import Matrix.LU
ppoly p x = map (x**) p
polyfit d ry = elems $ solve mat vec where
mat = listArray ((1,1), (d,d)) $ liftM2 concatMap ppoly id [0..fromIntegral $ pred d]
vec = listArray (1,d) $ take d ry