open Owl

module R = Regression.D

let input =
  Arr.uniform ~a:0.0 ~b:10.0 [| 30; 1 |]

let c = 400.0

let b = 0.0


let output =
  Arr.init [| 30; 1 |] (fun i ->
      (Arr.get input [| i; 0 |]) *. c +. b
    )

let w, bias =
  let res = R.ols ~i:true input output in
  res.(0) , res.(1)

(* 1x1 matrix = 59.62... *)
let _ =
  Arr.print w

(* 1x1 matrix = 59.96... *)
let _ =
  Arr.print bias