- PyTorch (GPU Enabled)
- PyTorch Lightning
- numpy
- pandas
- matplotlib
- seaborn
- Install required packages
- run preprosses.py to generate dataset/input.pt and dataset/target.pt files
- run main.py with or without arguments (Detailed arguments are given in main.py file)
- run main.py with --test argument enabled to generate prediction as dataset/#_pred.pt
- run postprosses.py to generate csv files