@ptrblck yes sir, I tried matplotlib
plt.plot(epochs, mean_loss, 'g', label='Training loss')
plt.plot(epochs, mean_acc, 'b', label='Training  acc')
plt.title('Training  loss and acc')
plt.xlabel('Epochs')
plt.ylabel('Loss and acc')
plt.legend()
plt.show()

but in my code (in initial question dont know how to use it) as above code in train and test function can plot loss and acc for training and testing. But I want to plot loss for train and test in a single plot and similarly acc for train and test.
Regards