package tensorflow

  1. Overview
  2. Docs
type 'a net = 'a t
type 'a t
val create : 'a net -> 'a t
val evaluate : ?named_inputs:(Input_name.t * (float, Bigarray.float32_elt) Tensor.t) list -> 'a t -> (float, Bigarray.float32_elt) Tensor.t -> (float, Bigarray.float32_elt) Tensor.t
module Optimizer : sig ... end
module Loss : sig ... end
val fit : ?named_inputs:(Input_name.t * (float, Bigarray.float32_elt) Tensor.t) list -> loss:Loss.t -> optimizer:Optimizer.t -> epochs:int -> xs:(float, Bigarray.float32_elt) Tensor.t -> ys:(float, Bigarray.float32_elt) Tensor.t -> 'a t -> unit
OCaml

Innovation. Community. Security.