How to include callbacks for plotting Training and Validation Loss Graphy?

[MHansy]

Hello all I am trying to use this repo., but I dont know how I can edit the codes so as (1) To plot the training and validation loss graphy

(2) To make prediction to get the predictions probability

Any help please

Thanks