Plot loss and accuracy with Module API

1

Hi,

Sorry for asking this basic question. When training, using the Module API, I’d like to store properties like loss, accuracy, etc during training. For instance, the Keras function fit returns a History dictionary containing these values. But I can’t find it how it is done with MXNet. I don’t assume that one needs to do complicated text parsing from the std.out (DEBUG) log.

Many thanks in advance,
Blake

2

You can use the Metric API to store and update your metrics: https://mxnet.incubator.apache.org/versions/master/api/python/metric/metric.html
Since you want to store multiple metrics you may want to use mxnet.metric. CompositeEvalMetric

1 Like