noggin.xarray.metrics_to_xarrays

noggin.xarray.metrics_to_xarrays(metrics: Dict[str, Dict[str, numpy.ndarray]]) → Tuple[xarray.core.dataset.Dataset, xarray.core.dataset.Dataset]

Given noggin metrics, returns xarray datasets for the batch-level and epoch-level metrics, respectively.

Parameters:

metrics : Dict[str, Dict[str, ndarray]]

Live metrics reported as a dictionary, (e.g. via LivePlot.train_metrics or LivePlot.test_metrics)

Returns:

MetricArrays[xarray.Dataset, xarray.Dataset]

The batch-level and epoch-level datasets. The metrics are reported as data variables in the dataset, and the coordinates corresponds to the batch-iteration count.

Notes

The layout of the resulting data sets are:

Dimensions:     (iterations: num_iterations)
Coordinates:
  * iterations  (iterations) int64 1 2 3 ...
Data variables:
    metric0      (iterations) float64 val_0 val_1 ...
    metric1      (iterations) float64 val_0 val_1 ...
    ...