plt.plot(accuracy_change, label = 'train')
plt.plot(test_accuracy, label = 'test')
plt.xlabel('Epoch')
plt.ylabel('Accuracy')
plt.ylim([0.884, 0.8877])
plt.legend(loc='lower right')
plt.show()

plt.plot(loss_change, label = 'train')
plt.plot(test_loss, label = 'test')
plt.xlabel('Epoch')
plt.ylabel('Loss')
plt.ylim([0.31, 0.33])
plt.legend(loc='lower right')