package tiny_libs

  1. Overview
  2. Docs

Module NetSource

Sourcetype activation =
  1. | Sigmoid
  2. | Tanh
  3. | Relu
  4. | Linear
Sourceval squash : activation -> float -> float
Sourceval slope : activation -> z:float -> a:float -> float
Sourcetype layer = {
  1. w : Matrix.t;
  2. b : Matrix.t;
  3. f : activation;
}
Sourcetype t = layer list
Sourceval make : seed:int -> ?hidden:activation -> ?last:activation -> int list -> t
Sourceval forward : t -> float array -> float array
Sourceval sizes : t -> int list
Sourceval weights : t -> int
Sourcetype pass = {
  1. input : Matrix.t;
  2. steps : (Matrix.t * Matrix.t) list;
}
Sourceval forward_pass : t -> float array -> pass
Sourceval output : pass -> float array