Module Owl_neural_optimise

module Owl_neural_optimise: sig .. end

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