-
Notifications
You must be signed in to change notification settings - Fork 0
ICP 11
In this lab we covered the deep learning through convolutional neural network. Convolutional neural networks are mainly used for image classification. For this lab we trained the model on CIFAR-10 dataset.
The tasks are as follows.
- Modify the model
- Predict the 4 classes and show it on plot
- Visualize loss and accuracy using history object
- Save the model and then predict the images used loaded model
The model is modified by adding multiple convolutional layers, drop outs, max pooling layer and fully connected layer. The summary of the model is given below.
The accuracy of the model before modifying the model is given below.
After the modification the accuracy of the model increased. It is shown below.
The model predicted the classes as shown below.
The loss and accuracy of the of the model is shown below.
Save the model using the following code
Load the model using the following screenshot
The following output was generated using the new model loaded.