hey @Joy-Gupta-2763139277246091 ,
Firstly , A really good notebook. Well done.
Secondly coming to your doubt , see improving a deep learning model doesn’t means that we need to always increase the number of layers in it.
It that was to be true , then we would be taking hundreds of layers to get it done.
As you are making it from scratch , in that case choosing the correct number of layers and deciding number of nodes in each layer plays a very important role.
Along it also matters how do you preprocess the input images and provide to your model.
Optimizing learning rate , number of epochs plays a very big role.
While dealing with images , in deep learning we use convolutional layers but to implement them from scratch is a quite tricky task and takes a lot of time.
You can try other models also , like KNN , or LogisticRegression . But no one can’t gurantee that which mode will work the best.
You need to try as much as you can and select the one that works best.
I hope this helps you
.