AI class: Linear Regression

LinearRegression.hs

-- https://www.ai-class.com/course/video/quizquestion/127

module LinearRegression where

computeLR :: [(Double, Double)] -> (Double, Double)
computeLR vs = (w0, w1)
  where w1 = enum / denom
        enum  = m * (sum $ zipWith (*) xs ys) - (sum xs) * (sum ys)
        denom = m * (sum . map (^2) $ xs) - (sum xs)^2
        w0 = (sum ys - w1 * sum xs) / m
        m = fromIntegral (length vs)
        xs = map fst vs
        ys = map snd vs