All the result of testing is available in main.ipynb file.
Augmentation.py is used for preprocessing data, enrich the original positive class because it was critically imbalanced.
For dataset, check folder Data included in this git.