pihnn.graphics.plot_loss

pihnn.graphics.plot_loss(loss_train, loss_test, figsize=None, format='png', dir='results/')

Plot of the training and test loss.

Parameters:

• loss_train (list of float) – Training loss at each epoch.

• loss_train – Test loss at each epoch.

• figsize (tuple of float) – Size of the matplotlib.pyplot.figure.

• format (str) – Format of the save figure.

• dir (str) – Directory where to save the figure.