package owl-base

  1. Overview
  2. Docs
package owl-base
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Source file owl_neural_generic.ml

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# 1 "src/base/neural/owl_neural_generic.ml"
(*
 * OWL - OCaml Scientific Computing
 * Copyright (c) 2016-2022 Liang Wang <liang@ocaml.xyz>
 *)

(** Functor to create neural networks of different precision. *)

module Make_Embedded (A : Owl_types_ndarray_algodiff.Sig) = struct
  include
    Owl_neural_graph.Make
      (Owl_neural_neuron.Make (Owl_optimise_generic.Make (Owl_algodiff_generic.Make (A))))
end

module Flatten (Graph : Owl_neural_graph_sig.Sig) = struct
  (* module aliases: graphical network & parallel *)

  module Graph = Graph
  module Optimise = Graph.Neuron.Optimise
  module Algodiff = Graph.Neuron.Optimise.Algodiff

  (* module Parallel       = Owl_neural_parallel.Make (Graph) *)

  (* module aliases: weight init and activation *)

  module Init = Graph.Neuron.Init
  module Activation = Graph.Neuron.Activation

  (* module aliases: optimisation configuration *)

  module Params = Graph.Neuron.Optimise.Params
  module Batch = Graph.Neuron.Optimise.Batch
  module Learning_Rate = Graph.Neuron.Optimise.Learning_Rate
  module Loss = Graph.Neuron.Optimise.Loss
  module Gradient = Graph.Neuron.Optimise.Gradient
  module Momentum = Graph.Neuron.Optimise.Momentum
  module Regularisation = Graph.Neuron.Optimise.Regularisation
  module Clipping = Graph.Neuron.Optimise.Clipping
  module Stopping = Graph.Neuron.Optimise.Stopping
  module Checkpoint = Graph.Neuron.Optimise.Checkpoint
end

module Make (A : Owl_types_ndarray_algodiff.Sig) = struct
  include Flatten (Make_Embedded (A))
end

(* ends here *)
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct