The accuracy and loss values after 10 epochs for training and validation data using plot(model_one) can be seen in the following graph:
From the preceding graph, the following observations can be made:
- The training loss continues to decrease from epoch 1 to 10.
- Validation loss reduces initially, but it starts to look flat after 3 epochs.
- A similar pattern is also observed for accuracy in the opposite direction.
In the next section, we will evaluate the classification model and assess model prediction performance with the help of train and test data.