matrix.ml

open Lacaml.D
open Lacaml.Io
open Str

type t = Mat.t

let mat_const m n c =
  Mat.random m n ~from:(c) ~range:(0.0)

let mat_random m n =
  Mat.random m n ~from:(-0.1) ~range:(0.2)

let mat_ones m n =
  Mat.random m n ~from:(1.0) ~range:(0.0)

let mat_zeros m n =
  Mat.random m n ~from:(0.0) ~range:(0.0)

let mat_const m n c =
  Mat.random m n ~from:(c) ~range:(0.0)

let to_list m =
  Mat.to_list m

let floats_to_str (l : float list) =
  let rec loop acc = function
    | [] -> acc
    | h::t -> if acc = "" then loop (acc ^ string_of_float h) t
              else loop (acc ^ " " ^ string_of_float h) t
  in loop "" l

let strs_to_str (l : string list) =
  let rec loop acc = function
    | [] -> acc
    | h::t -> if acc = "" then loop (acc ^ h ^ "\n") t
              else loop (acc ^ " " ^ h ^ "\n") t
in loop "" l

let save_weights path m =
  let file = open_out path in
  Printf.fprintf file "%s" (to_list m
                            |> List.map (fun x -> floats_to_str x)
                            |> strs_to_str);
  close_out file;
  path

(* [file_to_list filename] reads a file and creates a list of its contents,
 * where each line is a list within the larger list. *)
let file_to_list filename =
  let ch = open_in filename in
  let rec parse_file ch acc =
    match input_line ch with
    | exception End_of_file -> close_in ch; acc
    | line -> parse_file ch (Str.split (regexp " ") line::acc)
  in parse_file ch []

let load_weights path =
  try
    (List.rev (file_to_list path))
    |> List.map (fun x -> List.map (fun y -> float_of_string y) x)
    |> Mat.of_list
  with
  | _ -> failwith "error in loading from filepath"

let print m =
    let () =
        let a = m in
        let rows, cols = (Mat.dim1 a), (Mat.dim2 a) in
        Format.printf "@[<2>This is an indented random matrix:@\n@\n%a@]@."
        (Lacaml.Io.pp_lfmat
            ~row_labels:
            (Array.init rows (fun i -> Printf.sprintf "Row %d" (i + 1)))
            ~col_labels:
            (Array.init cols (fun i -> Printf.sprintf "Col %d" (i + 1)))
            ~vertical_context:(Some (Context.create 784))
            ~horizontal_context:(Some (Context.create 784))
            ~ellipsis:"*"
            ~print_right:false
            ~print_foot:false ())
        m in ()