add examples

1 files changed, 18 insertions, 0 deletions

diff --git a/examples/pinv1.hs b/examples/pinv1.hs
new file mode 100644
index 0000000..76fa0a9
--- /dev/null
+++ b/examples/pinv1.hs
@@ -0,0 +1,18 @@
+-- initial check for the polynomial model example
+import LinearAlgebra
+
+
+prepSyst :: Int -> Matrix Double -> (Matrix Double, Vector Double)
+prepSyst n d = (a,b) where
+    [x,b] = toColumns d
+    a = fromColumns $ 1+0*x : map (x^) [1 .. n]
+
+main = do
+    dat <- readMatrix `fmap` readFile "data.txt"
+    let (a,b) = prepSyst 3 dat
+    putStr "Coefficient matrix:\n"
+    dispR 3 a
+    putStr "Desired values:\n"
+    print b
+    putStr "\nLeast Squares Solution:\n"
+    print $ pinv a <> b -- equivalent to: linearSolveLSR a (fromColumns [b])