package tensorflow

You can search for identifiers within the package.

in-package search v0.2.0

tensorflow
- Tensorflow
  - Cell
  - Fnn
    
    Input_id
    
    Loss
    
    Model
    
    Optimizer
    
    Shape
  - Gradients
  - Layer
  - Mnist_helper
  - Node
    
    Attr_list
    
    Dim
    
    Tensor
    
    Type
    
    Weak_table
  - Ops
    
    Op_names
    
    Placeholder
  - Ops_generated
    
    Op_names
  - Ops_gradients
  - Ops_manual
    
    Placeholder
  - Optimizers
  - Registered_gradients
  - Session
    
    Input
    
    Output
  - Tensor
  - Var
  - Wrapper
    
    Graph
    
    Session
    
    Session_options
    
    Status

Legend:
Library
Module
Module type
Parameter
Class
Class type

type 'a fnn = 'a t

type ('a, 'b, 'c) t

val create : ('c * 'b) Tensor.eq -> 'a fnn -> ('a, 'b, 'c) t

val predict : 
  ('a, 'b, 'c) t ->
  (Input_id.t * (float, 'c) Tensor.t) list ->
  (float, 'c) Tensor.t

val fit : 
  ?addn_inputs:(Input_id.t * (float, 'c) Tensor.t) list ->
  ?batch_size:int ->
  ('a, 'b, 'c) t ->
  loss:Loss.t ->
  optimizer:Optimizer.t ->
  epochs:int ->
  input_id:Input_id.t ->
  xs:(float, 'c) Tensor.t ->
  ys:(float, 'c) Tensor.t ->
  unit

val save : ('a, 'b, 'c) t -> filename:string -> unit

val load : ('a, 'b, 'c) t -> filename:string -> unit