Project information
- Category: Machine Learning
- Project date: April 2020
- GitHub repo: link
About COVID-19 Classifier
I collaborated with three partners to tackle a private COVID-19 classification Kaggle competitition; our transfer learning-based model achieved a test accuracy of 87%. The final PyTorch model used VGG16 as a feature extractor with custom neural net layers and I implemented the image augmentation used to mitigate the small dataset size.