poplar.nn.plot
¶
Module Contents¶
Functions¶
|
Simple plot routine for producing a loss curve |
- 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.