From ba6ead6d7a41b7ed78bb228181b7262d0c75d2eb Mon Sep 17 00:00:00 2001
From: YurenHao0426 <blackhao0426@gmail.com>
Date: Mon, 4 May 2026 23:10:10 -0500
Subject: =?UTF-8?q?Global=20rename=20GRAFT=20=E2=86=92=20KAFT=20(incl.=20i?=
 =?UTF-8?q?nternal=20class=20+=20filenames)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 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.
---
 src/trainers.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'src/trainers.py')

diff --git a/src/trainers.py b/src/trainers.py
index 651dffc..7589fc9 100644
--- a/src/trainers.py
+++ b/src/trainers.py
@@ -1,5 +1,5 @@
 """
-Training methods for Graph-GrAPE experiments.
+Training methods for KAFT experiments.
 Generalized to L-layer residual GCN.
 
 Methods compared:
@@ -7,7 +7,7 @@ Methods compared:
   DFA          — Fixed random R, P=I
   DFA-GNN      — Fixed random R, P=Â^{L-l}
   VanillaGrAPE — Aligned R (per layer), P=I
-  GraphGrAPE   — Aligned R (per layer) + topology P=Â^{L-l}
+  KAFT   — Aligned R (per layer) + topology P=Â^{L-l}
 """
 
 import torch
@@ -179,7 +179,7 @@ class BPTrainer:
 # ---------------------------------------------------------------------------
 
 class _FeedbackTrainerBase:
-    """Shared logic for DFA / GrAPE variants, generalized to L layers."""
+    """Shared logic for DFA / KAFT variants, generalized to L layers."""
 
     def __init__(self, data, hidden_dim, lr, weight_decay,
                  diffusion_alpha, diffusion_iters,
@@ -608,10 +608,10 @@ class VanillaGrAPETrainer(_FeedbackTrainerBase):
 
 
 # ---------------------------------------------------------------------------
-# Graph-GrAPE Trainer
+# KAFT Trainer
 # ---------------------------------------------------------------------------
 
-class GraphGrAPETrainer(_FeedbackTrainerBase):
+class KAFTTrainer(_FeedbackTrainerBase):
     """Aligned R per layer + topology P = Â^{min(L-l, max_power)}."""
 
     def __init__(self, data, hidden_dim, lr, weight_decay,
-- 
cgit v1.2.3