Skip to content

saniyanafees6/Automated-Diagnosis-Model-Using-Transfer-Learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Automated-Diagnosis-Model-Using-Transfer-Learning

This is my first attempt at using transfer learning. The task at hand was to train a machine learning model on Kaggle's Pneumonia Data Set, to accurately predict if the patient has Pneumonia or not.

In order to achieve this, I used InceptionV3 as the base model, and performed transfer learning with added layers, to improve performance.

I was able to achieve accurate results as can be seen in the confusion matrix below:

🎉 I had a lot of fun working on this project, and learned a lot along the way! 🎉

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors