owl 0.2.5 · OCaml Package

owl

module Feedforward = Owl_neural_layer.Feedforward

module Init = Owl_neural_layer.Init

module Activation = Owl_neural_layer.Activation

module Params = Owl_neural_optimise.Params

module Batch = Owl_neural_optimise.Batch

module Learning_Rate = Owl_neural_optimise.Learning_Rate

module Loss = Owl_neural_optimise.Loss

module Gradient = Owl_neural_optimise.Gradient

module Momentum = Owl_neural_optimise.Momentum

module Regularisation = Owl_neural_optimise.Regularisation

module Clipping = Owl_neural_optimise.Clipping

val input : int array -> Owl_neural_layer.layer

val activation : Activation.typ -> Owl_neural_layer.layer

val linear : ?init_typ:Init.typ -> ?inputs:int -> int -> Owl_neural_layer.layer

val linear_nobias : 
  ?init_typ:Init.typ ->
  ?inputs:int ->
  int ->
  Owl_neural_layer.layer

val recurrent : 
  ?init_typ:Init.typ ->
  act_typ:Owl_neural_layer.Activation.typ ->
  ?inputs:int ->
  int ->
  int ->
  Owl_neural_layer.layer

val lstm : ?inputs:int -> int -> Owl_neural_layer.layer

val gru : ?inputs:int -> int -> Owl_neural_layer.layer

val conv2d : 
  ?padding:Owl_algodiff.S.padding ->
  ?inputs:int array ->
  int array ->
  int array ->
  Owl_neural_layer.layer

val conv3d : 
  ?padding:Owl_algodiff.S.padding ->
  ?inputs:int array ->
  'a ->
  int array ->
  int array ->
  Owl_neural_layer.layer

val fully_connected : 
  ?init_typ:Init.typ ->
  ?inputs:int ->
  int ->
  Owl_neural_layer.layer

val max_pool2d : 
  ?padding:Owl_algodiff.S.padding ->
  int array ->
  int array ->
  Owl_neural_layer.layer

val avg_pool2d : 
  ?padding:Owl_algodiff.S.padding ->
  int array ->
  int array ->
  Owl_neural_layer.layer

val dropout : float -> Owl_neural_layer.layer

val reshape : 
  ?conv_typ:bool ->
  ?inputs:int array ->
  int array ->
  Owl_neural_layer.layer

val flatten : ?conv_typ:bool -> unit -> Owl_neural_layer.layer

val lambda : 
  (Owl_neural_layer.t -> Owl_neural_layer.t) ->
  Owl_neural_layer.layer

val train : 
  ?params:Owl_neural_optimise.Params.typ ->
  Owl_neural_layer.network ->
  Owl_algodiff.S.mat ->
  Owl_algodiff.S.mat ->
  Owl_algodiff.S.elt array

val test_model : 
  Owl_neural_layer.network ->
  Owl_algodiff.S.mat ->
  Owl_algodiff.S.mat ->
  unit

val train_cnn : 
  ?params:Owl_neural_optimise.Params.typ ->
  Owl_neural_layer.network ->
  Owl_algodiff.S.arr ->
  Owl_algodiff.S.mat ->
  Owl_algodiff.S.elt array

val print : Owl_neural_layer.network -> unit

val save : 'a -> string -> unit

val load : string -> Owl_neural_layer.network

package owl