Hi everyone, model accuracy is heavily dependent on your epochs and learning rates. I’ve achieved 99.25% accuracy in my model after tweaking hyper-parameters a few times. I gave an attempt to explain why in my medium blog post here. You can also find my notebook here. I will probably experiment the use of neural networks on pneumonia image classification so stay tuned throughout the rest of the week.
EDIT: As stated below from @aakashns, there was a reference error in the dataloader. For anyone that reviewed my notebook and blog post, I apologize. I should’ve been a bit more unsettled since the model was showing the same accuracy for both training and testing. The new accuracy from my model is around 55% which is much more understandable since the model is only looking at each pixel rather than the full picture. I assume that this is why we have convolution networks. Again, I apologize for any misunderstandings. I’ve updated my notebook here and updated my blog post here.