tensorflow

TensorFlow bindings for OCaml
IN THIS PACKAGE
Module Tensorflow_fnn . Fnn . Model
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