package kaun

  1. Overview
  2. Docs
package kaun
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Metrics.CollectionSource

Sourcetype t

Layout-agnostic collection of named metrics.

Sourceval empty : unit -> t
Sourceval of_list : (string * metric) list -> t
Sourceval create : (string * metric) list -> t
Sourceval add : t -> string -> metric -> unit
Sourceval remove : t -> string -> unit
Sourceval reset : t -> unit
Sourceval update : t -> predictions:(float, 'layout) Rune.t -> targets:(_, 'layout) Rune.t -> ?loss:(float, 'layout) Rune.t -> ?weights:(float, 'layout) Rune.t -> unit -> unit
Sourceval compute : t -> (string * float) list
Sourceval compute_tensors : t -> (string * Ptree.tensor) list
Sourceval compute_dict : t -> (string, float) Hashtbl.t