Pneumonia prediction using transfer learning
Train Loss: 0.3667
Train Accuracy: 0.8405
Test Loss: 0.6079217195510864
Test Accuracy: 0.7308443188667297
Train Loss: 0.2763555943965912
Train Accuracy: 0.8705905079841614
Test Loss: 0.3666607737541199
Test Accuracy: 0.8445512652397156
Train Loss: 0.11729855090379715
Train Accuracy: 0.9555214643478394
Test loss: 0.19975192844867706
Test Accuracy: 0.9198718070983887
After fine tuning the ResNet50 model
Train Loss: 0.06547978520393372
Train Accuracy: 0.9779524803161621
Test loss: 0.16307932138442993
Test Accuracy: 0.9439102411270142