From d12722525fc010a3910b5152c72654a2ade5eac4 Mon Sep 17 00:00:00 2001
From: YurenHao0426 <Blackhao0426@gmail.com>
Date: Wed, 17 Jun 2026 11:19:27 -0500
Subject: Initial import

---
 diag/run_diag.sh | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 diag/run_diag.sh

(limited to 'diag/run_diag.sh')

diff --git a/diag/run_diag.sh b/diag/run_diag.sh
new file mode 100644
index 0000000..4a6f31a
--- /dev/null
+++ b/diag/run_diag.sh
@@ -0,0 +1,16 @@
+#!/usr/bin/env bash
+# Step-1 diagnosis sweep. CSL = pure-H2 anchor (classification).
+# Triangle counting on ER (graphs 1-WL-distinguishable -> H2~0, the H1 end) and
+# regular (graphs collapse to one WL color -> H2~var, the ceiling end).
+set -uo pipefail
+cd /home/yurenh2/rrog
+export PYTHONPATH=/home/yurenh2/rrog
+echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)"
+run() { echo "===== $* ====="; python3 diag/train_diag.py "$@" --layers 4 --epochs 300 --seed 0 \
+        || echo "!! FAILED $*"; }
+for model in gin gcn; do
+  run --task csl --model "$model"
+  run --task tri --model "$model" --kind er
+  run --task tri --model "$model" --kind regular
+done
+echo "done=$(date -Is)"
-- 
cgit v1.2.3