summaryrefslogtreecommitdiff
path: root/experiments/run_cora_perturb.py
diff options
context:
space:
mode:
authorYurenHao0426 <blackhao0426@gmail.com>2026-05-04 23:10:10 -0500
committerYurenHao0426 <blackhao0426@gmail.com>2026-05-04 23:10:10 -0500
commitba6ead6d7a41b7ed78bb228181b7262d0c75d2eb (patch)
tree726171fb4b0c536d9287a15daf52929ec65fa3d0 /experiments/run_cora_perturb.py
parent37ba0f83e3652a215680fd8515af9c14fc02e21c (diff)
Global rename GRAFT → KAFT (incl. internal class + filenames)
- src/trainers.py: GraphGrAPETrainer → KAFTTrainer; module docstring + comments. VanillaGrAPETrainer kept as-is (it is a separate control method, not KAFT). - experiments/: all 19 runners pick up the new class name; result keys ('Cora_GRAFT' etc) become 'Cora_KAFT'; OUT_DIRs renamed (e.g. bp_graft_depth_20seeds → bp_kaft_depth_20seeds). - figures/: data-lookup keys + display labels both 'KAFT'; output filename graft_depth_sweep.{pdf,png} → kaft_depth_sweep.{pdf,png}. - File rename: experiments/run_bp_graft_depth.py → run_bp_kaft_depth.py; figures/graft_depth_sweep.pdf → kaft_depth_sweep.pdf. - README aligned. Imports verified: from src.trainers import KAFTTrainer succeeds.
Diffstat (limited to 'experiments/run_cora_perturb.py')
-rw-r--r--experiments/run_cora_perturb.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/experiments/run_cora_perturb.py b/experiments/run_cora_perturb.py
index 1dabc95..1124b97 100644
--- a/experiments/run_cora_perturb.py
+++ b/experiments/run_cora_perturb.py
@@ -13,7 +13,7 @@ import numpy as np
import json
import os
from src.data import load_dataset, build_normalized_adj, build_row_normalized_adj
-from src.trainers import BPTrainer, DFATrainer, GraphGrAPETrainer
+from src.trainers import BPTrainer, DFATrainer, KAFTTrainer
device = 'cuda:0'
SEEDS = [0, 1, 2, 3, 4]
@@ -111,7 +111,7 @@ def main():
common = dict(data=data, hidden_dim=64, lr=0.01, weight_decay=5e-4,
num_layers=L, residual_alpha=0.0, backbone='gcn')
bp_accs.append(train_one(BPTrainer, common, {}, seed))
- gr_accs.append(train_one(GraphGrAPETrainer, common, grape_extra, seed))
+ gr_accs.append(train_one(KAFTTrainer, common, grape_extra, seed))
bp, gr = np.mean(bp_accs)*100, np.mean(gr_accs)*100
delta = gr - bp