Project website: https://coldcode0214.github.io/CS539/
CS539.ipynb contains our main source code. It should be run in Google CoLab.
Our code depends on packages such as numpy, pandas, pytorch and sklearn.
The following Google Drive share contains the dataset we used to train, test and validate our models. It was downsampled from a Kaggle dataset of lung CT images.
https://drive.google.com/drive/folders/1PddWQdPVhiRDITNUptk-1yrtZS_ZE2qL