kaun
Kaun.Metrics
kaun.datasets
type t
type metric
val avg : string -> metric
val sum : string -> metric
val accuracy : string -> metric
val create : metric list -> t
val update : t -> ?loss:('layout, 'dev) tensor -> ?logits:('layout, 'dev) tensor -> ?labels:('layout, 'dev) tensor -> unit -> unit
val compute : t -> (string * float) list
val get : t -> string -> float
val reset : t -> unit