package tensorflow

  1. Overview
  2. Docs
type float32_genarray = (float, Bigarray.float32_elt, Bigarray.c_layout) Bigarray.Genarray.t
type t = {
  1. train_images : float32_genarray;
  2. train_labels : float32_genarray;
  3. test_images : float32_genarray;
  4. test_labels : float32_genarray;
}
val read_files : ?train_image_file:string -> ?train_label_file:string -> ?test_image_file:string -> ?test_label_file:string -> unit -> t
val train_batch : t -> batch_size:int -> batch_idx:int -> float32_genarray * float32_genarray
val image_dim : int
val label_count : int
val slice2 : ('a, 'b, 'c) Bigarray.Genarray.t -> int -> int -> ('a, 'b, Bigarray.c_layout) Bigarray.Genarray.t
val accuracy : float32_genarray -> float32_genarray -> float
OCaml

Innovation. Community. Security.