Let's now draw a plot of the training and testing losses during the training process. ChatGPT explained that I need to change the line that fits the model to capture its result value (history.) That was neat! The plot works and looks good!