Legend:
Library
Module
Module type
Parameter
Class
Class type
Library
Module
Module type
Parameter
Class
Class type
type 'a fnn = 'a 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