Source file `owl_regression_generic_sig.ml`

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
# 1 "src/owl/optimise/owl_regression_generic_sig.ml"
(*
 * OWL - OCaml Scientific and Engineering Computing
 * Copyright (c) 2016-2019 Liang Wang <liang.wang@cl.cam.ac.uk>
 *)

module type Sig = sig

  module Optimise : Owl_optimise_generic_sig.Sig

  open Optimise.Algodiff


  (** {6 Type definition} *)

  type arr = A.arr
  (** Type of ndarray values. *)

  type elt = A.elt
  (** Type of scalar values. *)


  (** {6 Regression models} *)

  val ols : ?i:bool -> arr -> arr -> arr array
  (** TODO *)

  val ridge : ?i:bool -> ?alpha:float -> arr -> arr -> arr array
  (** TODO *)

  val lasso : ?i:bool -> ?alpha:float -> arr -> arr -> arr array
  (** TODO *)

  val elastic_net : ?i:bool -> ?alpha:float -> ?l1_ratio:float -> arr -> arr -> arr array
  (** TODO *)

  val svm : ?i:bool -> ?a:float -> arr -> arr -> arr array
  (** TODO *)

  val logistic : ?i:bool -> arr -> arr -> arr array
  (** TODO *)

  val exponential : ?i:bool -> arr -> arr -> elt * elt * elt
  (** TODO *)

  val poly : arr -> arr -> int -> arr
  (** TODO *)


end