package tensorflow

  1. Overview
  2. Docs
type 'a fnn = 'a t
type ('a, 'b, 'c) t
val create : ('c * 'b) Tensorflow_core.Tensor.eq -> 'a fnn -> ('a, 'b, 'c) t
val predict : ('a, 'b, 'c) t -> ?output_id:Id.t -> (Input_id.t * (Base.float, 'c) Tensorflow_core.Tensor.t) Base.list -> (Base.float, 'c) Tensorflow_core.Tensor.t
val fit : ?addn_inputs: (Input_id.t * (Base.float, 'c) Tensorflow_core.Tensor.t) Base.list -> ?batch_size:Base.int -> ?explicit_vars:'a fnn Base.list -> ('a, 'b, 'c) t -> loss:Loss.t -> optimizer:Optimizer.t -> epochs:Base.int -> input_id:Input_id.t -> xs:(Base.float, 'c) Tensorflow_core.Tensor.t -> ys:(Base.float, 'c) Tensorflow_core.Tensor.t -> Base.unit
val save : ?inputs:(Input_id.t * (Base.float, 'c) Tensorflow_core.Tensor.t) Base.list -> ('a, 'b, 'c) t -> filename:Base.string -> Base.unit
val load : ?inputs:(Input_id.t * (Base.float, 'c) Tensorflow_core.Tensor.t) Base.list -> ('a, 'b, 'c) t -> filename:Base.string -> Base.unit
OCaml

Innovation. Community. Security.