I am done with plotting the graph. But my problem is I don\'t know how to write a code that can compute and display the loss and accuracy of the trained model on the test se