module Owl_neural_optimise: sig
.. end
Neural network: optimisation engine
module Utils: sig
.. end
module Learning_Rate: sig
.. end
module Batch: sig
.. end
module Loss: sig
.. end
module Gradient: sig
.. end
module Momentum: sig
.. end
module Regularisation: sig
.. end
module Clipping: sig
.. end
module Stopping: sig
.. end
module Params: sig
.. end
val _print_info : int -> int -> Owl_algodiff.S.t -> Owl_algodiff.S.t -> unit
val _print_summary : float -> unit
val train_nn : Params.typ ->
(Owl_algodiff.S.t -> Owl_algodiff.S.t * Owl_algodiff.S.t array array) ->
(Owl_algodiff.S.t ->
Owl_algodiff.S.t array array * Owl_algodiff.S.t array array) ->
(Owl_algodiff.S.t array array -> 'a) ->
(string -> unit) ->
Owl_algodiff.S.t -> Owl_algodiff.S.t -> Owl_algodiff.S.elt array
val train_nn_generic : ?params:Params.typ ->
('a -> Owl_algodiff.S.t -> Owl_algodiff.S.t * Owl_algodiff.S.t array array) ->
('a ->
Owl_algodiff.S.t ->
Owl_algodiff.S.t array array * Owl_algodiff.S.t array array) ->
('a -> Owl_algodiff.S.t array array -> 'b) ->
('a -> string -> unit) ->
'a -> Owl_algodiff.S.t -> Owl_algodiff.S.t -> Owl_algodiff.S.elt array
val test_nn_generic : (Owl_algodiff.S.mat -> unit) ->
(Owl_algodiff.S.t -> 'a -> Owl_algodiff.S.t) ->
'a -> Owl_algodiff.S.t -> Owl_algodiff.S.t -> unit