- Note: This repo is still under construction
- Link to the paper.
- Link to the data
src/trainer.py
: main handler for training.src/model.py
: model definitions.src/preprocessing.py
: data preparation steps, including DataFrame transformation, dataset spliting, and dataloader creation.
- See
requirements.txt
- Train:
python run.py -r -m CONFIG_FILE
- Test / Inference:
python run.py -s test
- Codebase: Zhaoyi H., Yifei N.
- Research / Paper: Xiaocheng G., Jingbo S., Vish K.