|
(import
|
|
[numpy [ones column-stack]]
|
|
[numpy.random [randn]]
|
|
[numpy.linalg [lstsq]])
|
|
|
|
(setv n 1000)
|
|
(setv x1 (randn n))
|
|
(setv x2 (randn n))
|
|
(setv y (+ 3 (* 1 x1) (* -2 x2) (* .25 x1 x2) (randn n)))
|
|
|
|
(print (first (lstsq
|
|
(column-stack (, (ones n) x1 x2 (* x1 x2)))
|
|
y)))
|