Code to replicate results in this paper: https://arxiv.org/pdf/2103.11002.pdf
-
Create a directory called "data/" in the root of this repo
-
Download the dataset here
-
Unzip into the "data/" directory
-
Create a virtualenv, and install the packages in requirements.txt
-
Create the adversarial samples by executing run_attacks.py, in src/data/
-
Train each of the 3 models in the paper, by running the following in src/models/
python train.py co_occur GPU_ID
python train.py laplace GPU_ID
python train.py direct GPU_ID
-
Produce test results by running this for all 3 methods:
python predict.py {co_occur,laplace,direct} GPU_ID
-
Print out results and confusion matrices as arrays and latex code:
python eval.py {co_occur,laplace,direct}