| # summarize the result and plot the training and test loss | |
| plt.plot(result.history['loss']) | |
| plt.plot(result.history['val_loss']) | |
| # Set the parameters | |
| plt.title('Deep learning model loss') | |
| plt.ylabel('Loss') | |
| plt.xlabel('Epochs') | |
| plt.legend(['train', 'test'], loc='upper right') | |
| # Display the plots | |
| plt.show() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment