poplar.nn.plot

Plotting routines for visualising network outputs and reporting training/testing diagnostics.

Functions

loss_plot(train_losses, test_losses[, filename])

Simple plot routine for producing a loss curve

Module Contents

poplar.nn.plot.loss_plot(train_losses: list, test_losses: list, filename=None)

Simple plot routine for producing a loss curve

Parameters:

train_losseslist

List of training data losses

test_losseslist

List of testing data losses

filenamestr, optional

Output string for the loss curve figure, by default None (in which case, the figure is returned).

Returns:

fig (only returned if filename is None)

The loss curve matplotlib figure.