Make (owl-base.Owl_optimise

Parameters

module Algodiff : Owl_algodiff_generic_sig.Sig

Signature

module Algodiff = Algodiff

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 Checkpoint : sig ... end

module Params : sig ... end

val minimise_weight : 
  ?state:Checkpoint.state ->
  Params.typ ->
  (Algodiff.t -> Algodiff.t -> Algodiff.t) ->
  Algodiff.t ->
  Algodiff.t ->
  Algodiff.t ->
  Checkpoint.state * Algodiff.t

val minimise_network : 
  ?state:Checkpoint.state ->
  Params.typ ->
  (Algodiff.t -> Algodiff.t * Algodiff.t array array) ->
  (Algodiff.t -> Algodiff.t array array * Algodiff.t array array) ->
  (Algodiff.t array array -> unit) ->
  (string -> unit) ->
  Algodiff.t ->
  Algodiff.t ->
  Checkpoint.state

val minimise_fun : 
  ?state:Checkpoint.state ->
  Params.typ ->
  (Algodiff.t -> Algodiff.t) ->
  Algodiff.t ->
  Checkpoint.state * Algodiff.t

val minimise_compiled_network : 
  ?state:Checkpoint.state ->
  Params.typ ->
  (Algodiff.t -> Algodiff.t -> Algodiff.t) ->
  (unit -> unit) ->
  (string -> unit) ->
  Algodiff.t ->
  Algodiff.t ->
  Checkpoint.state