package tiny_libs

  1. Overview
  2. Docs

Module TrainSource

Sourcetype history = {
  1. epoch : int;
  2. training : float;
  3. held_out : float;
}
Sourceval split : ?part:float -> Backprop.example list -> Backprop.example list * Backprop.example list
Sourceval epoch : ?seed:int -> ?rate:float -> ?batch:int -> Net.t -> Backprop.example list -> Net.t
Sourceval run : ?epochs:int -> ?rate:float -> ?decay:float -> ?batch:int -> ?seed:int -> ?held:Backprop.example list -> Net.t -> Backprop.example list -> Net.t * history list
Sourceval accuracy : Net.t -> Backprop.example list -> answer:(float array -> int) -> float
Sourceval best : float array -> int
Sourceval one_hot : int -> int -> float array