package caisar

You can search for identifiers within the package.

in-package search v0.2.0

Legend:
Library
Module
Module type
Parameter
Class
Class type

Module to parse neural networks written in the NNet format https://github.com/sisl/NNet

type t = private {

n_layers : int;
(*
Number of layers.
*)
n_inputs : int;
(*
Number of inputs.
*)
n_outputs : int;
(*
Number of outputs.
*)
max_layer_size : int;
(*
Maximum layer size.
*)
layer_sizes : int list;
(*
Size of each layer.
*)
min_input_values : float list option;
(*
Minimum values of inputs.
*)
max_input_values : float list option;
(*
Maximum values of inputs.
*)
mean_values : (float list * float) option;
(*
Mean values of inputs and one value for all outputs.
*)
range_values : (float list * float) option;
(*
Range values of inputs and one value for all outputs.
*)
weights_biases : float list list;
(*
All weights and biases of NNet model.
*)
nir : Nir.Ngraph.t;

}

NNet model metadata.

val parse : ?permissive:bool -> string -> (t, string) Result.t

Parse an NNet file.

parameter permissive
false by default. When set, parsing does not fail on non available information, which are set to None instead.

val to_nir : t -> Nir.Ngraph.t

Convert an well-formed NNet into a Nir.