README

Start here

Library

Reference

Tutorials

Other

public Metricsstruct

Metrics(metrics...)

Callback that tracks metrics during training.

metrics can be both AbstractMetrics or functions like f(ŷs, ys) which will be converted to Metrics.

A metric tracking lossfn is included by default.

Examples

  • metrics = Metrics(accuracy)
  • metrics = Metrics(Metric(Flux.mse, device = gpu), Metric(Flux.mae, device = gpu))