├── .gitignore ├── LICENSE.md ├── Makefile ├── README.md ├── code_documentation.md ├── examples_src └── pandemic │ ├── README.md │ ├── sir_model.py │ ├── solution_centralized.py │ └── solution_federated.py ├── local_run ├── centralized.py ├── centralized.sh └── centralized_test.py ├── runtime ├── Dockerfile ├── apt.txt ├── entrypoint.sh ├── environment-cpu.yml ├── environment-gpu.yml ├── main_centralized_test.py ├── main_centralized_train.py ├── main_federated_test.py ├── main_federated_train.py ├── supervisor.py └── toprc └── submission_src └── pandemic ├── accounting.py ├── data_utils.py ├── graph_utils.py ├── install.sh ├── loss_utils.py ├── solution_centralized.py ├── solution_federated.py └── training_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/LICENSE.md -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/README.md -------------------------------------------------------------------------------- /code_documentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/code_documentation.md -------------------------------------------------------------------------------- /examples_src/pandemic/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/examples_src/pandemic/README.md -------------------------------------------------------------------------------- /examples_src/pandemic/sir_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/examples_src/pandemic/sir_model.py -------------------------------------------------------------------------------- /examples_src/pandemic/solution_centralized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/examples_src/pandemic/solution_centralized.py -------------------------------------------------------------------------------- /examples_src/pandemic/solution_federated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/examples_src/pandemic/solution_federated.py -------------------------------------------------------------------------------- /local_run/centralized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/local_run/centralized.py -------------------------------------------------------------------------------- /local_run/centralized.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/local_run/centralized.sh -------------------------------------------------------------------------------- /local_run/centralized_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/local_run/centralized_test.py -------------------------------------------------------------------------------- /runtime/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/runtime/Dockerfile -------------------------------------------------------------------------------- /runtime/apt.txt: -------------------------------------------------------------------------------- 1 | sysstat 2 | tree 3 | -------------------------------------------------------------------------------- /runtime/entrypoint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/runtime/entrypoint.sh -------------------------------------------------------------------------------- /runtime/environment-cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/runtime/environment-cpu.yml -------------------------------------------------------------------------------- /runtime/environment-gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/runtime/environment-gpu.yml -------------------------------------------------------------------------------- /runtime/main_centralized_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/runtime/main_centralized_test.py -------------------------------------------------------------------------------- /runtime/main_centralized_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/runtime/main_centralized_train.py -------------------------------------------------------------------------------- /runtime/main_federated_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/runtime/main_federated_test.py -------------------------------------------------------------------------------- /runtime/main_federated_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/runtime/main_federated_train.py -------------------------------------------------------------------------------- /runtime/supervisor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/runtime/supervisor.py -------------------------------------------------------------------------------- /runtime/toprc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/runtime/toprc -------------------------------------------------------------------------------- /submission_src/pandemic/accounting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/submission_src/pandemic/accounting.py -------------------------------------------------------------------------------- /submission_src/pandemic/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/submission_src/pandemic/data_utils.py -------------------------------------------------------------------------------- /submission_src/pandemic/graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/submission_src/pandemic/graph_utils.py -------------------------------------------------------------------------------- /submission_src/pandemic/install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/submission_src/pandemic/install.sh -------------------------------------------------------------------------------- /submission_src/pandemic/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/submission_src/pandemic/loss_utils.py -------------------------------------------------------------------------------- /submission_src/pandemic/solution_centralized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/submission_src/pandemic/solution_centralized.py -------------------------------------------------------------------------------- /submission_src/pandemic/solution_federated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/submission_src/pandemic/solution_federated.py -------------------------------------------------------------------------------- /submission_src/pandemic/training_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kenziyuliu/pets-challenge/HEAD/submission_src/pandemic/training_utils.py --------------------------------------------------------------------------------