I am playing with the code listed here by Daniel Persson on Youtube. His code at github.
I am playing with his code for my classification project and I got accuracy of about 88% (I am using a GPU). BUt I got about 93% with InceptionV3 and RasNEt50 transfer learning. I am new to ML and I managed to setup basic training models using Keras. I am using 3 classes (120x120 pxl RGB images). In the above code, I could not find how to change cross-entropy to categorial-cross-entropy.
What are the other methods to improve the accuracy level? I feel the output should be better since images differences are trivial to humans.
- Will this improve increasing hidden layers ?
- A number of nodes in existing layers ?
Also I would like to know how I could use sklearn kit to plot confusion matrix here.
Thank you in advance.