Project Creation: Part Three

In this lesson, we will perform some fine-tuning of the model and observe the accuracy of our model.

Until now, we have built a model using the pre-trained model as a feature Extractor and connected our own classifier on top of that. This is one of the strategies of transfer learning. Before we move to the fine-tuning, let’s try to do some predictions from the model_new we built in the previous lesson.

Make predictions

To make predictions, we would require a sample image to be loaded. Here, we have taken a Pikachu image; and let’s see what output we will get.

Get hands-on with 1200+ tech skills courses.