In this notebook, I treid to classify the digitst 5-9 by using a model that is trained on the digits 0-4 by emplying a techinque called transfer learning.
I used two approaches.
-
save the optimized weights from the initial model (trained on the digits 0-4) and use them to classify the digitst 5-9 which the model had never seen. In this technique rather than initilize the weights randomly I used the pretrained weights
-
The second apprach is to freez the first layers of the model, which usually is called feature layers. Then the model only trains, the last classification layes in classifying the digits 5-9.
Results:
