ICLR2021-Workshop-Challenge-Track2-Baseline

Requirement

numpy pandas matplotlib sklearn xgboost

Usage

For symptom predict

python symptom_train.py

For disease predict

python disease_train.py

The model will be saved in a file

For test and submit

python test.py

Then, the file "ans.pk" can be submmited.