package owl

  1. Overview
  2. Docs
package owl
Legend:
Library
Module
Module type
Parameter
Class
Class type

Parameters

module M : ModelSig
module E : EngineSig

Signature

type task = {
  1. mutable id : int;
  2. mutable params : Owl_neural_optimise.Params.typ;
  3. mutable model : M.network;
  4. mutable data_x : Owl_algodiff.S.t;
  5. mutable data_y : Owl_algodiff.S.t;
}
val delta_model : M.network -> M.network -> unit
val local_model : task -> 'a
val schedule : task -> 'a list -> ('a * (int * 'b) list) list
val pull : task -> ('a * M.network) list -> ('a * 'b) list
val push : task -> 'a -> ('b * M.network) list -> ('b * M.network) list
val stop : 'a -> 'b -> bool
val train_generic : ?params:Owl_neural_optimise.Params.typ -> M.network -> Owl_algodiff.S.t -> Owl_algodiff.S.t -> string -> string -> unit
val train : ?params:Owl_neural_optimise.Params.typ -> M.network -> Owl_algodiff.S.mat -> Owl_algodiff.S.mat -> string -> string -> unit
val train_cnn : ?params:Owl_neural_optimise.Params.typ -> M.network -> Owl_algodiff.S.arr -> Owl_algodiff.S.mat -> string -> string -> unit