add reproduce info

iaBIH · May 6, 2024 · 0e0203c · 0e0203c
1 parent f378c51
commit 0e0203c
Show file tree

Hide file tree

Showing 5 changed files with 10,161 additions and 0 deletions.
diff --git a/reproduce/README.md b/reproduce/README.md
@@ -0,0 +1,8 @@
+## Reproducing repository
+
+`run_txgnn.sh` and `train.py`: training scripts
+`gather result-txgnn.ipynb`: notebook to produce the plots in the paper
+`result_more_metrics.csv`: raw result files
+
+Model checkpoint here: https://drive.google.com/file/d/1BPHKejmUpERhLTY4negB4e6ZLmoE7dwq/view?usp=sharing
+You can use `TxGNN.load_pretrained` to load these checkpoints and evaluate.