src.ml.utils.plot_loss_acc_curves

plot_loss_acc_curves(loss_stats, acc_stats, model_name)

Plot loss and accuracy curves.

Parameters

• loss_stats – A dictionary with loss values and keys = “train” and “val”.

• acc_stats – A dictionary with accuracy values and keys = “train” and “val”.

• model_name – Name of model.