This notebook was a project done as part of my Caltech training.
The project goal was to create and train deep-learning models for classification of lung diseases from x-ray images
It includes:
- Data preprocessing steps (loading data, creating train/validation/test sets, descriptive stats, data augmentation)
- Training multiple CNN models
- Assessing performance of each (accuracy + loss between training/validation sets)
- Running each trained model on the test set
Enjoy!
Ryan